| /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 …]
|
| /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 | hlo_alias_analysis_test.cc | 1027 Shape elem_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
1030 HloInstruction::CreateParameter(0, elem_shape, "param")); in TEST_F()
1032 HloInstruction::CreateUnary(elem_shape, HloOpcode::kNegate, param0)); in TEST_F()
1034 HloInstruction::CreateUnary(elem_shape, HloOpcode::kNegate, negate)); in TEST_F()
1051 Shape elem_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
1054 HloInstruction::CreateParameter(0, elem_shape, "param")); in TEST_F()
1056 HloInstruction::CreateUnary(elem_shape, HloOpcode::kNegate, param0)); in TEST_F()
1058 HloInstruction::CreateUnary(elem_shape, HloOpcode::kNegate, negate)); in TEST_F()
|
| D | tuple_points_to_analysis_test.cc | 879 Shape elem_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
881 0, ShapeUtil::MakeTupleShape({elem_shape, elem_shape}), "tuple")); in TEST_F()
883 HloInstruction::CreateGetTupleElement(elem_shape, tuple, 0)); in TEST_F()
885 HloInstruction::CreateGetTupleElement(elem_shape, tuple, 1)); in TEST_F()
887 HloInstruction::CreateBinary(elem_shape, HloOpcode::kAdd, gte0, gte1)); in TEST_F()
|
| D | while_loop_simplifier.cc | 871 const auto& elem_shape = while_shape.tuple_shapes(i); in TryMergeInductionVariables() local
877 elem_shape, HloOpcode::kAdd, add_gte(instr, i), in TryMergeInductionVariables()
878 add_binary_op(elem_shape, HloOpcode::kMultiply, in TryMergeInductionVariables()
|
| D | buffer_assignment_test.cc | 1565 auto elem_shape = f32vec4_; in TEST_F() local
1566 auto tuple_shape = ShapeUtil::MakeTupleShape({elem_shape}); in TEST_F()
1570 HloInstruction::CreateParameter(0, elem_shape, "sub_param")); in TEST_F()
1577 HloInstruction::CreateParameter(0, elem_shape, "param")); in TEST_F()
1606 auto elem_shape = f32vec4_; in TEST_F() local
1607 auto tuple_shape = ShapeUtil::MakeTupleShape({elem_shape}); in TEST_F()
1630 HloInstruction::CreateParameter(0, elem_shape, "param")); in TEST_F()
|
| D | hlo_dataflow_analysis_test.cc | 2042 Shape elem_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2044 0, ShapeUtil::MakeTupleShape({elem_shape, elem_shape}), "tuple")); in TEST_F()
2046 HloInstruction::CreateGetTupleElement(elem_shape, tuple, 0)); in TEST_F()
2048 HloInstruction::CreateGetTupleElement(elem_shape, tuple, 1)); in TEST_F()
2050 HloInstruction::CreateBinary(elem_shape, HloOpcode::kAdd, gte0, gte1)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 474 for (const Shape& elem_shape : shape.tuple_shapes()) { in HumanString() local
475 StrAppend(&text, prefix, HumanString(elem_shape)); in HumanString()
498 for (const Shape& elem_shape : shape.tuple_shapes()) { in HumanStringWithLayout() local
499 StrAppend(&text, prefix, HumanStringWithLayout(elem_shape)); in HumanStringWithLayout()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | lower_static_tensor_list.cc | 598 auto elem_shape = rewriter->create<TF::SliceOp>( in CreateCondTrueBranch() local
602 loc, resize_op.output_handle().getType(), elem_shape, size_diff); in CreateCondTrueBranch()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 355 TF_ASSIGN_OR_RETURN(xla::Shape elem_shape, in BuildComputation()
357 elem_shapes.push_back(elem_shape); in BuildComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2521 const Shape& elem_shape = in HandleCustomCall() local
2523 TF_RET_CHECK(!elem_shape.IsTuple()) << "Nested tuples not implemented"; in HandleCustomCall()
2526 llvm::Value* addr = EmitBufferPointer(slice, elem_shape); in HandleCustomCall()
|