Home
last modified time | relevance | path

Searched refs:elem_shape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_array_ops.cc52 const TensorShape& elem_shape) { in MaybeInitializeTensorArray() argument
70 TF_RETURN_IF_ERROR(resource->SetTypeAndShape(dtype, elem_shape)); in MaybeInitializeTensorArray()
84 ta_shape.AppendShape(elem_shape); in MaybeInitializeTensorArray()
207 TensorShape elem_shape = ctx->InputShape(2); in Compile() local
214 MaybeInitializeTensorArray(b, resource, dtype_, elem_shape)); in Compile()
224 xla::MakeEdgePaddingConfig({{0, elem_shape.dims()}})); in Compile()
226 TensorShape slice_shape = elem_shape; in Compile()
369 TensorShape elem_shape = value_shape; in Compile() local
370 elem_shape.RemoveDim(0); in Compile()
372 MaybeInitializeTensorArray(b, resource, dtype_, elem_shape)); in Compile()
[all …]
Dstack_ops.cc65 const TensorShape& elem_shape) { in MaybeInitializeStack() argument
74 stack_shape.AppendShape(elem_shape); in MaybeInitializeStack()
80 TF_RETURN_IF_ERROR(resource->SetTypeAndShape(dtype, elem_shape)); in MaybeInitializeStack()
142 TensorShape elem_shape = ctx->InputShape(1); in Compile() local
148 OP_REQUIRES_OK(ctx, MaybeInitializeStack(b, resource, dtype_, elem_shape)); in Compile()
157 xla::MakeEdgePaddingConfig({{0, elem_shape.dims()}})); in Compile()
159 TensorShape slice_shape = elem_shape; in Compile()
/external/tensorflow/tensorflow/compiler/xla/service/
Dliveness_util_test.cc57 Shape elem_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
59 0, ShapeUtil::MakeTupleShape({elem_shape, elem_shape}), "tuple")); in TEST_F()
61 HloInstruction::CreateGetTupleElement(elem_shape, tuple, 0)); in TEST_F()
63 HloInstruction::CreateGetTupleElement(elem_shape, tuple, 1)); in TEST_F()
65 HloInstruction::CreateBinary(elem_shape, HloOpcode::kAdd, gte0, gte1)); in TEST_F()
Dbuffer_assignment_test.cc1221 auto elem_shape = f32vec4_; in TEST_F() local
1222 auto tuple_shape = ShapeUtil::MakeTupleShape({elem_shape}); in TEST_F()
1226 HloInstruction::CreateParameter(0, elem_shape, "sub_param")); in TEST_F()
1233 HloInstruction::CreateParameter(0, elem_shape, "param")); in TEST_F()
1260 auto elem_shape = f32vec4_; in TEST_F() local
1261 auto tuple_shape = ShapeUtil::MakeTupleShape({elem_shape}); in TEST_F()
1284 HloInstruction::CreateParameter(0, elem_shape, "param")); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc392 for (const Shape& elem_shape : shape.tuple_shapes()) { in HumanString() local
393 tensorflow::strings::StrAppend(&text, prefix, HumanString(elem_shape)); in HumanString()
459 for (const Shape& elem_shape : shape.tuple_shapes()) { in HumanStringWithLayout() local
461 HumanStringWithLayout(elem_shape)); in HumanStringWithLayout()