Searched refs:gradients_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | boosted_trees_ops.cc | 322 shape_inference::ShapeHandle gradients_shape; in __anon770f674c0902() local
328 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &gradients_shape)); in __anon770f674c0902()
331 c->Dim(gradients_shape, 0), &unused_dim)); in __anon770f674c0902()
333 c->Merge(gradients_shape, hessians_shape, &unused_shape)); in __anon770f674c0902()
364 shape_inference::ShapeHandle gradients_shape; in __anon770f674c0a02() local
371 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &gradients_shape)); in __anon770f674c0a02()
376 TF_RETURN_IF_ERROR(c->Merge(c->Dim(gradients_shape, 0), in __anon770f674c0a02()
414 shape_inference::ShapeHandle gradients_shape; in __anon770f674c0b02() local
424 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &gradients_shape)); in __anon770f674c0b02()
431 TF_RETURN_IF_ERROR(c->Merge(c->Dim(gradients_shape, 0), in __anon770f674c0b02()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | pooling_ops.cc | 425 TensorShape gradients_shape; in Compile() local
426 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &gradients_shape)); in Compile()
431 OP_REQUIRES(ctx, gradients_shape.dims() == num_dims(), in Compile()
451 XlaTensorFormat(data_format_, gradients_shape.dims() - 2); in Compile()
453 MakeSpatialPadding(gradients_shape.dim_sizes(), ksize_, stride_int64s, in Compile()
456 xla::AvgPoolGrad(converted_out_backprop, gradients_shape.dim_sizes(), in Compile()
|