Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dsparse_to_dense_op.cc89 TensorShape output_tensor_shape; in Compute() local
/external/tensorflow/tensorflow/core/kernels/linalg/
Dlinalg_ops_common.cc182 TensorShape output_tensor_shape({}); in PrepareOutputs() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.cc407 TensorShape input_tensor_shape, filter_tensor_shape, output_tensor_shape; in MakeXlaBackpropFilterConvOp() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc2745 TensorShapeProto* input_tensor_shape, TensorShapeProto* output_tensor_shape, in IsReductionCandidateForSimplification()
2840 TensorShapeProto input_tensor_shape, output_tensor_shape; in SimplifyReduction() local