Searched refs:output_tensor_shape (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_to_dense_op.cc | 115 TensorShape output_tensor_shape; in Compute() local
118 &output_tensor_shape)); in Compute()
120 OP_REQUIRES_OK(c, c->allocate_output(0, output_tensor_shape, &output)); in Compute()
219 TensorShape output_tensor_shape; in ComputeAsync() local
223 &output_tensor_shape), in ComputeAsync()
226 OP_REQUIRES_OK_ASYNC(c, c->allocate_output(0, output_tensor_shape, &output), in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | linalg_ops_common.cc | 182 TensorShape output_tensor_shape({}); in PrepareOutputs() local
195 output_tensor_shape = batch_shape; in PrepareOutputs()
196 output_tensor_shape.AppendShape(output_matrix_shape); in PrepareOutputs()
204 input_idx, output_idx, output_tensor_shape, &out)) { in PrepareOutputs()
213 output_idx, output_tensor_shape, &out)); in PrepareOutputs()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | gradients.py | 49 output_tensor_shape = output.shape
77 out.set_shape(output_tensor_shape.concatenate(flat_inputs[i].shape))
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 407 TensorShape input_tensor_shape, filter_tensor_shape, output_tensor_shape; in MakeXlaBackpropFilterConvOp() local
410 TF_RETURN_IF_ERROR(XLAShapeToTensorShape(output_shape, &output_tensor_shape)); in MakeXlaBackpropFilterConvOp()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.h | 230 TensorShapeProto* output_tensor_shape, bool* is_single_element_op) const;
|
| D | constant_folding.cc | 2761 TensorShapeProto* input_tensor_shape, TensorShapeProto* output_tensor_shape, in IsReductionCandidateForSimplification() argument
2776 *output_tensor_shape = output_props.shape(); in IsReductionCandidateForSimplification()
2782 for (int i = 0; i < output_tensor_shape->dim_size(); ++i) { in IsReductionCandidateForSimplification()
2783 if (output_tensor_shape->dim(i).size() < 0) { in IsReductionCandidateForSimplification()
2790 TensorShape(*output_tensor_shape).num_elements(); in IsReductionCandidateForSimplification()
2856 TensorShapeProto input_tensor_shape, output_tensor_shape; in SimplifyReduction() local
2858 *node, properties, &input_tensor_shape, &output_tensor_shape, in SimplifyReduction()
2888 const int new_num_dimensions = output_tensor_shape.dim_size(); in SimplifyReduction()
|