| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | stack_ops.cc | 63 DataType dtype, const TensorShape& elem_shape) { in MaybeInitializeStack() argument
72 stack_shape.AppendShape(elem_shape); in MaybeInitializeStack()
76 TF_RETURN_IF_ERROR(resource->SetTypeAndShape(dtype, elem_shape)); in MaybeInitializeStack()
135 TensorShape elem_shape = ctx->InputShape(1); in Compile() local
141 OP_REQUIRES_OK(ctx, MaybeInitializeStack(b, resource, dtype_, elem_shape)); in Compile()
148 std::vector<xla::XlaOp> start_indices(elem_shape.dims() + 1, in Compile()
152 TensorShape slice_shape = elem_shape; in Compile()
|
| D | tensor_array_ops.cc | 52 const TensorShape& elem_shape) { in MaybeInitializeTensorArray() argument
67 TF_RETURN_IF_ERROR(resource->SetTypeAndShape(dtype, elem_shape)); in MaybeInitializeTensorArray()
81 ta_shape.AppendShape(elem_shape); in MaybeInitializeTensorArray()
200 TensorShape elem_shape = ctx->InputShape(2); in Compile() local
207 MaybeInitializeTensorArray(b, resource, dtype_, elem_shape)); in Compile()
215 std::vector<xla::XlaOp> start_indices(elem_shape.dims() + 1, in Compile()
219 TensorShape slice_shape = elem_shape; in Compile()
368 TensorShape elem_shape = value_shape; in Compile() local
369 elem_shape.RemoveDim(0); in Compile()
371 MaybeInitializeTensorArray(b, resource, dtype_, elem_shape)); in Compile()
[all …]
|
| D | tensor_list_ops.cc | 351 TensorShape elem_shape = ctx->InputShape(2); in Compile() local
354 OP_REQUIRES_OK(ctx, GetInitializedList(tl, elem_shape, dtype_, &buffer)); in Compile()
362 std::vector<xla::XlaOp> start_indices(elem_shape.dims() + 1, in Compile()
366 TensorShape slice_shape = elem_shape; in Compile()
394 TensorShape elem_shape = ctx->InputShape(1); in Compile() local
398 GetInitializedList(list_tuple, elem_shape, dtype_, &buffer)); in Compile()
405 std::vector<xla::XlaOp> start_indices(elem_shape.dims() + 1, in Compile()
409 TensorShape slice_shape = elem_shape; in Compile()
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | slices_test.py | 31 elem_shape = constant_op.constant([2])
32 l = list_ops.tensor_list_from_tensor(initial_list, element_shape=elem_shape)
41 elem_shape = constant_op.constant([2])
42 l = list_ops.tensor_list_from_tensor(initial_list, element_shape=elem_shape)
|
| D | data_structures_test.py | 125 elem_shape = constant_op.constant([2])
126 l = list_ops.tensor_list_from_tensor(initial_list, element_shape=elem_shape)
150 elem_shape = constant_op.constant([2])
151 l = list_ops.tensor_list_from_tensor(initial_list, element_shape=elem_shape)
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tuple_points_to_analysis_test.cc | 853 Shape elem_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
855 0, ShapeUtil::MakeTupleShape({elem_shape, elem_shape}), "tuple")); in TEST_F()
857 HloInstruction::CreateGetTupleElement(elem_shape, tuple, 0)); in TEST_F()
859 HloInstruction::CreateGetTupleElement(elem_shape, tuple, 1)); in TEST_F()
861 HloInstruction::CreateBinary(elem_shape, HloOpcode::kAdd, gte0, gte1)); in TEST_F()
|
| D | while_loop_simplifier.cc | 870 const auto& elem_shape = while_shape.tuple_shapes(i); in TryMergeInductionVariables() local
876 elem_shape, HloOpcode::kAdd, add_gte(instr, i), in TryMergeInductionVariables()
877 add_binary_op(elem_shape, HloOpcode::kMultiply, in TryMergeInductionVariables()
|
| D | buffer_assignment_test.cc | 1386 auto elem_shape = f32vec4_; in TEST_F() local
1387 auto tuple_shape = ShapeUtil::MakeTupleShape({elem_shape}); in TEST_F()
1391 HloInstruction::CreateParameter(0, elem_shape, "sub_param")); in TEST_F()
1398 HloInstruction::CreateParameter(0, elem_shape, "param")); in TEST_F()
1427 auto elem_shape = f32vec4_; in TEST_F() local
1428 auto tuple_shape = ShapeUtil::MakeTupleShape({elem_shape}); in TEST_F()
1451 HloInstruction::CreateParameter(0, elem_shape, "param")); in TEST_F()
|
| D | hlo_dataflow_analysis_test.cc | 1969 Shape elem_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
1971 0, ShapeUtil::MakeTupleShape({elem_shape, elem_shape}), "tuple")); in TEST_F()
1973 HloInstruction::CreateGetTupleElement(elem_shape, tuple, 0)); in TEST_F()
1975 HloInstruction::CreateGetTupleElement(elem_shape, tuple, 1)); in TEST_F()
1977 HloInstruction::CreateBinary(elem_shape, HloOpcode::kAdd, gte0, gte1)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 427 for (const Shape& elem_shape : shape.tuple_shapes()) { in HumanString() local
428 StrAppend(&text, prefix, HumanString(elem_shape)); in HumanString()
451 for (const Shape& elem_shape : shape.tuple_shapes()) { in HumanStringWithLayout() local
452 StrAppend(&text, prefix, HumanStringWithLayout(elem_shape)); in HumanStringWithLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2278 const Shape& elem_shape = in HandleCustomCall() local
2280 TF_RET_CHECK(!elem_shape.IsTuple()) << "Nested tuples not implemented"; in HandleCustomCall()
2283 llvm::Value* addr = EmitBufferPointer(slice, elem_shape); in HandleCustomCall()
|