Searched refs:output_tensor_shape (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | linalg_ops_common.cc | 180 TensorShape output_tensor_shape({}); in PrepareOutputs() local
193 output_tensor_shape = batch_shape; in PrepareOutputs()
194 output_tensor_shape.AppendShape(output_matrix_shape); in PrepareOutputs()
202 input_idx, output_idx, output_tensor_shape, &out)) { in PrepareOutputs()
211 output_idx, output_tensor_shape, &out)); in PrepareOutputs()
|
| D | sparse_to_dense_op.cc | 89 TensorShape output_tensor_shape; in Compute() local
92 &output_tensor_shape)); in Compute()
94 OP_REQUIRES_OK(c, c->allocate_output(0, output_tensor_shape, &output)); in Compute()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | gradients.py | 47 output_tensor_shape = output.shape
75 out.set_shape(output_tensor_shape.concatenate(flat_inputs[i].shape))
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 410 TensorShape input_tensor_shape, filter_tensor_shape, output_tensor_shape; in MakeXlaBackpropFilterConvOp() local
413 TF_RETURN_IF_ERROR(XLAShapeToTensorShape(output_shape, &output_tensor_shape)); in MakeXlaBackpropFilterConvOp()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.h | 160 TensorShapeProto* output_tensor_shape, bool* is_single_element_op) const;
|
| D | constant_folding.cc | 2465 TensorShapeProto* input_tensor_shape, TensorShapeProto* output_tensor_shape, in IsReductionCandidateForSimplification() argument
2490 *output_tensor_shape = output_props.shape(); in IsReductionCandidateForSimplification()
2496 for (int i = 0; i < output_tensor_shape->dim_size(); ++i) { in IsReductionCandidateForSimplification()
2497 if (output_tensor_shape->dim(i).size() < 0) { in IsReductionCandidateForSimplification()
2504 TensorShape(*output_tensor_shape).num_elements(); in IsReductionCandidateForSimplification()
2545 TensorShapeProto input_tensor_shape, output_tensor_shape; in SimplifyReduction() local
2547 *node, properties, &input_tensor_shape, &output_tensor_shape, in SimplifyReduction()
2577 const int new_num_dimensions = output_tensor_shape.dim_size(); in SimplifyReduction()
|