Home
last modified time | relevance | path

Searched defs:elem_shape (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstack_ops.cc63 DataType dtype, const TensorShape& elem_shape) { in MaybeInitializeStack()
135 TensorShape elem_shape = ctx->InputShape(1); in Compile() local
Dtensor_array_ops.cc52 const TensorShape& elem_shape) { in MaybeInitializeTensorArray()
200 TensorShape elem_shape = ctx->InputShape(2); in Compile() local
368 TensorShape elem_shape = value_shape; in Compile() local
506 TensorShape elem_shape = value_shape; in Compile() local
Dtensor_list_ops.cc351 TensorShape elem_shape = ctx->InputShape(2); in Compile() local
394 TensorShape elem_shape = ctx->InputShape(1); in Compile() local
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc427 for (const Shape& elem_shape : shape.tuple_shapes()) { in HumanString() local
451 for (const Shape& elem_shape : shape.tuple_shapes()) { in HumanStringWithLayout() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_simplifier.cc870 const auto& elem_shape = while_shape.tuple_shapes(i); in TryMergeInductionVariables() local
Dtuple_points_to_analysis_test.cc853 Shape elem_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
Dbuffer_assignment_test.cc1386 auto elem_shape = f32vec4_; in TEST_F() local
1427 auto elem_shape = f32vec4_; in TEST_F() local
Dhlo_dataflow_analysis_test.cc1969 Shape elem_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2278 const Shape& elem_shape = in HandleCustomCall() local