Home
last modified time | relevance | path

Searched defs:output_tensor_shape (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_to_dense_op.cc115 TensorShape output_tensor_shape; in Compute() local
221 TensorShape output_tensor_shape; in ComputeAsync() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dlinalg_ops_common.cc189 TensorShape output_tensor_shape({}); in PrepareOutputs() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dconv_op_helpers.cc407 TensorShape input_tensor_shape, filter_tensor_shape, output_tensor_shape; in MakeXlaBackpropFilterConvOp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dconstant_folding.cc2795 TensorShapeProto* input_tensor_shape, TensorShapeProto* output_tensor_shape, in IsReductionCandidateForSimplification()
2890 TensorShapeProto input_tensor_shape, output_tensor_shape; in SimplifyReduction() local