Searched refs:gradients_shape (Results 1 - 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/contrib/boosted_trees/ops/ |
H A D | split_handler_ops.cc | 51 ShapeHandle gradients_shape; 52 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(3), 1, &gradients_shape)); 54 c->Dim(gradients_shape, 0), &unused_dim)); 110 ShapeHandle gradients_shape; 111 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(3), 1, &gradients_shape)); 113 c->Dim(gradients_shape, 0), &unused_dim)); 169 ShapeHandle gradients_shape; 170 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(3), 1, &gradients_shape)); 172 c->Dim(gradients_shape, 0), &unused_dim));
|
H A D | stats_accumulator_ops.cc | 77 ShapeHandle gradients_shape; 79 c->input(num_resource_handles * 3 + i + 1), 1, &gradients_shape)); 81 c->Dim(gradients_shape, 0), &unused_dim)); 163 ShapeHandle gradients_shape; 164 TF_RETURN_IF_ERROR(c->WithRank(c->input(5), 1, &gradients_shape)); 166 c->Dim(gradients_shape, 0), &unused_dim)); 295 ShapeHandle gradients_shape; 297 c->input(num_resource_handles * 3 + i + 1), 2, &gradients_shape)); 299 c->Dim(gradients_shape, 0), &unused_dim)); 382 ShapeHandle gradients_shape; [all...] |
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
H A D | stats_accumulator_ops.cc | 276 TensorShape gradients_shape = gradients_t.shape(); local 281 gradients_shape.RemoveDim(0); 286 context, gradients_shape == accumulator_resource->gradient_shape(), 288 "Gradients dimensions must match: ", gradients_shape.DebugString(), 305 std::vector<float> new_gradients(gradients_shape.num_elements()); 306 for (int j = 0; j < gradients_shape.num_elements(); ++j) { 316 for (int j = 0; j < gradients_shape.num_elements(); ++j) { 773 TensorShape gradients_shape = gradients_t->shape(); variable 774 gradients_shape.RemoveDim(0); 782 new StatsAccumulatorTensorResource(gradients_shape, hessians_shap [all...] |
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | pooling_ops.cc | 409 TensorShape gradients_shape; variable 410 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &gradients_shape)); 415 OP_REQUIRES(ctx, gradients_shape.dims() == num_dims(), 455 gradients_shape, filter_shape, out_backprop_shape, stride_, 472 ctx, out_backprop, dtype, gradients_shape, xla_padding, ksize_,
|
Completed in 450 milliseconds