Searched defs:elem_shape (Results 1 – 9 of 9) sorted by relevance
63 DataType dtype, const TensorShape& elem_shape) { in MaybeInitializeStack()135 TensorShape elem_shape = ctx->InputShape(1); in Compile() local
52 const TensorShape& elem_shape) { in MaybeInitializeTensorArray()200 TensorShape elem_shape = ctx->InputShape(2); in Compile() local368 TensorShape elem_shape = value_shape; in Compile() local506 TensorShape elem_shape = value_shape; in Compile() local
351 TensorShape elem_shape = ctx->InputShape(2); in Compile() local394 TensorShape elem_shape = ctx->InputShape(1); in Compile() local
427 for (const Shape& elem_shape : shape.tuple_shapes()) { in HumanString() local451 for (const Shape& elem_shape : shape.tuple_shapes()) { in HumanStringWithLayout() local
870 const auto& elem_shape = while_shape.tuple_shapes(i); in TryMergeInductionVariables() local
853 Shape elem_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
1386 auto elem_shape = f32vec4_; in TEST_F() local1427 auto elem_shape = f32vec4_; in TEST_F() local
1969 Shape elem_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2278 const Shape& elem_shape = in HandleCustomCall() local