Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/fuzzing/
Dscatter_nd_fuzz.cc66 auto flat_shape = shape_tensor.flat<int32>(); in FuzzImpl() local
/external/tensorflow/tensorflow/core/grappler/costs/
Danalytical_cost_estimator_test.cc71 auto flat_shape = ops::Const(s.WithOpName("flat_shape"), in CreateMiniGraph() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_simplifier.cc683 const Shape& flat_shape = instr->shape(); in TryFlattenNestedTuples() local
/external/tensorflow/tensorflow/core/kernels/
Ddecode_proto_op.cc789 TensorShape flat_shape = {static_cast<int64>(message_count), in Compute() local