Searched refs:gradients_shape (Results 1 - 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/ops/
H A Dsplit_handler_ops.cc51 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 Dstats_accumulator_ops.cc77 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 Dstats_accumulator_ops.cc276 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 Dpooling_ops.cc409 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