Home
last modified time | relevance | path

Searched refs:out_backprop_shape (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_shape_utils.cc96 const TensorShape& filter_shape, const TensorShape& out_backprop_shape, in ConvBackpropComputeDimensionsV2() argument
110 if (out_backprop_shape.dims() != num_dims) { in ConvBackpropComputeDimensionsV2()
116 if (dims->batch_size != out_backprop_shape.dim_size(batch_dim)) { in ConvBackpropComputeDimensionsV2()
120 ", outbackprop batch: ", out_backprop_shape.dim_size(batch_dim), in ConvBackpropComputeDimensionsV2()
135 if (dims->out_depth != out_backprop_shape.dim_size(feature_dim)) { in ConvBackpropComputeDimensionsV2()
148 label, input_shape, filter_shape, out_backprop_shape, dilations, in ConvBackpropComputeDimensionsV2()
158 const TensorShape& out_backprop_shape, in ConvBackpropComputeDimensions() argument
164 label, num_spatial_dims, input_shape, filter_shape, out_backprop_shape, in ConvBackpropComputeDimensions()
171 const TensorShape& out_backprop_shape, in Conv2DBackpropComputeInputShape() argument
185 const int batch_size = GetTensorDim(out_backprop_shape, data_format, 'N'); in Conv2DBackpropComputeInputShape()
Dconv_grad_shape_utils.h72 const TensorShape& out_backprop_shape,
82 const TensorShape& filter_shape, const TensorShape& out_backprop_shape,
90 const TensorShape& out_backprop_shape,
Dconv_grad_ops_3d.cc230 const TensorShape& out_backprop_shape = out_backprop.shape(); in Compute() local
245 input_shape, filter_shape, out_backprop_shape, in Compute()
337 const TensorShape& out_backprop_shape = out_backprop.shape(); in Compute() local
352 input_shape, filter_shape, out_backprop_shape, in Compute()
427 out_backprop_shape.num_elements(); in Compute()
688 const TensorShape& out_backprop_shape = out_backprop.shape(); in Compute() local
703 input_shape, filter_shape, out_backprop_shape, stride_, in Compute()
800 const TensorShape& out_backprop_shape = out_backprop.shape(); in Compute() local
815 input_shape, filter_shape, out_backprop_shape, stride_, in Compute()
889 out_backprop_shape.num_elements(); in Compute()
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv2d_backprop_filter_grad_test.py57 out_backprop_shape = conv_out.get_shape().as_list()
59 2 * np.random.random_sample(out_backprop_shape) - 1,
68 [in_val, out_backprop_val], [in_shape, out_backprop_shape],
101 out_backprop_shape = conv_out.get_shape().as_list()
103 2 * np.random.random_sample(out_backprop_shape) - 1,
113 [in_val, out_backprop_val], [in_shape, out_backprop_shape],
Dconv3d_backprop_filter_v2_grad_test.py51 out_backprop_shape = conv_out.get_shape().as_list()
53 2 * np.random.random_sample(out_backprop_shape) - 1,
59 [in_val, out_backprop_val], [in_shape, out_backprop_shape],
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbias_ops.cc91 const TensorShape out_backprop_shape = ctx->InputShape(0); in Compile() local
93 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrixOrHigher(out_backprop_shape), in Compile()
95 out_backprop_shape.DebugString())); in Compile()
99 ? out_backprop_shape.dims() - 1 in Compile()
106 std::vector<int64> reduce_dims(out_backprop_shape.dims() - 1); in Compile()
Dconv_op_helpers.cc138 const xla::Shape& filter_shape, const xla::Shape& out_backprop_shape, in ConvBackpropComputeDimensionsV2XlaShapes() argument
147 XLAShapeToTensorShape(out_backprop_shape, &out_backprop_tensor_shape)); in ConvBackpropComputeDimensionsV2XlaShapes()
305 TF_ASSIGN_OR_RETURN(xla::Shape out_backprop_shape, in MakeXlaBackpropInputConvOp()
320 out_backprop_shape, attrs.dilations, attrs.strides, attrs.padding, in MakeXlaBackpropInputConvOp()
346 if (out_backprop_shape.is_dynamic_dimension(dim)) { in MakeXlaBackpropInputConvOp()
400 TF_ASSIGN_OR_RETURN(xla::Shape out_backprop_shape, in MakeXlaBackpropFilterConvOp()
405 xla::Shape output_shape = out_backprop_shape; in MakeXlaBackpropFilterConvOp()
424 grouped_filter_shape, out_backprop_shape, attrs.dilations, attrs.strides, in MakeXlaBackpropFilterConvOp()
Dpooling_ops.cc317 const TensorShape out_backprop_shape = ctx->InputShape(2); in Compile() local
327 OP_REQUIRES(ctx, out_backprop_shape.dims() == num_dims(), in Compile()
349 OP_REQUIRES(ctx, expected_out_shape == out_backprop_shape, in Compile()
428 const TensorShape out_backprop_shape = ctx->InputShape(1); in Compile() local
436 OP_REQUIRES(ctx, out_backprop_shape.dims() == num_dims(), in Compile()
537 const TensorShape out_backprop_shape = ctx->InputShape(2); in Compile() local
547 OP_REQUIRES(ctx, out_backprop_shape.dims() == num_dims(), in Compile()
/external/tensorflow/tensorflow/compiler/tests/
Dconv3d_test.py53 out_backprop_shape = conv_out.get_shape().as_list()
55 2 * np.random.random_sample(out_backprop_shape) - 1,
61 [in_val, out_backprop_val], [in_shape, out_backprop_shape],
/external/tensorflow/tensorflow/python/profiler/internal/
Dflops_registry.py341 out_backprop_shape = graph_util.tensor_shape_from_node_def_name(graph,
343 out_backprop_shape.assert_is_fully_defined()
350 kernel_area * out_backprop_shape.num_elements() * 2)
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc4303 ArrayRef<int64_t> out_backprop_shape = out_backprop_ty.getShape(); in matchAndRewrite() local
4342 ToTensorShape<int64_t, num_dims>(out_backprop_shape), dilations, in matchAndRewrite()