Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtensor_list_ops_decomposition.mlir8 %elem_shape = "tf.Const"() {value = dense<> : tensor<0xi32>} : () -> tensor<0xi32>
16 …%tl = "tf.EmptyTensorList"(%elem_shape, %max_size) : (tensor<0xi32>, tensor<i32>) -> tensor<!tf.va…
33 …%pop:2 = "tf.TensorListPopBack"(%push, %elem_shape) : (tensor<!tf.variant<tensor<f32>>>, tensor<0x…
50 %elem_shape = "tf.Const"() {value = dense<> : tensor<0xi32>} : () -> tensor<0xi32>
59 …%tl = "tf.TensorListReserve"(%elem_shape, %num) : (tensor<0xi32>, tensor<i32>) -> tensor<!tf.varia…
74 …%get = "tf.TensorListGetItem"(%set, %arg0, %elem_shape) : (tensor<!tf.variant<tensor<f32>>>, tenso…
81 …%stack = "tf.TensorListStack"(%addn2, %elem_shape) : (tensor<!tf.variant<tensor<f32>>>, tensor<0xi…
96 %elem_shape = "tf.Const"() {value = dense<> : tensor<0xi32>} : () -> tensor<0xi32>
99 …%tl = "tf.TensorListFromTensor"(%arg1, %elem_shape) : (tensor<10xf32>, tensor<0xi32>) -> tensor<!t…
106 …%get = "tf.TensorListGetItem"(%tl, %arg0, %elem_shape) : (tensor<!tf.variant<tensor<f32>>>, tensor…
[all …]
Dshape_inference.mlir484 %elem_shape = "tf.Const"() {value = dense<[-1, 1]> : tensor<2xi32>} : () -> tensor<2xi32>
489 …%tl_0 = "tf.EmptyTensorList"(%elem_shape, %size) : (tensor<2xi32>, tensor<i32>) -> tensor<!tf.vari…
492 …%stack = "tf.TensorListStack"(%tl_2, %elem_shape) {num_elements = -1 : i64} : (tensor<!tf.variant<…
498 %elem_shape = "tf.Const"() {value = dense<[-1, 1]> : tensor<2xi32>} : () -> tensor<2xi32>
504 …%tl_0 = "tf.EmptyTensorList"(%elem_shape, %size) : (tensor<2xi32>, tensor<i32>) -> tensor<!tf.vari…
512 %elem_shape = "tf.Const"() {value = dense<[-1, 1]> : tensor<2xi32>} : () -> tensor<2xi32>
518 …%tl_0 = "tf.EmptyTensorList"(%elem_shape, %size) : (tensor<2xi32>, tensor<i32>) -> tensor<!tf.vari…
526 %elem_shape = "tf.Const"() {value = dense<[-1, 1]> : tensor<2xi32>} : () -> tensor<2xi32>
532 …%tl_0 = "tf.EmptyTensorList"(%elem_shape, %size) : (tensor<2xi32>, tensor<i32>) -> tensor<!tf.vari…
541 %elem_shape = "tf.Const"() {value = dense<[-1, 1]> : tensor<2xi32>} : () -> tensor<2xi32>
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstack_ops.cc63 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()
Dtensor_array_ops.cc52 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/
Dslices_test.py31 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)
Ddata_structures_test.py126 elem_shape = constant_op.constant([2])
127 l = list_ops.tensor_list_from_tensor(initial_list, element_shape=elem_shape)
151 elem_shape = constant_op.constant([2])
152 l = list_ops.tensor_list_from_tensor(initial_list, element_shape=elem_shape)
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtensor_list_ops_decomposition.cc757 TF::TensorListElementShapeOp elem_shape, in HandleTensorListElementShapeOp() argument
759 if (buffer_to_size.count(elem_shape.input_handle()) == 0) { in HandleTensorListElementShapeOp()
760 return elem_shape.emitOpError("unknown tensor list"); in HandleTensorListElementShapeOp()
762 auto buffer = elem_shape.input_handle(); in HandleTensorListElementShapeOp()
765 OpBuilder(elem_shape), elem_shape.getLoc(), in HandleTensorListElementShapeOp()
766 elem_shape.shape_type().getIntOrFloatBitWidth()); in HandleTensorListElementShapeOp()
767 elem_shape.element_shape().replaceAllUsesWith(result); in HandleTensorListElementShapeOp()
768 elem_shape.erase(); in HandleTensorListElementShapeOp()
859 } else if (auto elem_shape = in DecomposeTensorListOpsInternal() local
861 if (failed(HandleTensorListElementShapeOp(elem_shape, *buffer_to_size))) { in DecomposeTensorListOpsInternal()
Dtensor_array_ops_decomposition.cc96 llvm::SmallVector<int64_t, 8> elem_shape; in GetSplitElementTypeAndCount() local
97 elem_shape.push_back(length); in GetSplitElementTypeAndCount()
99 elem_shape.push_back(dim); in GetSplitElementTypeAndCount()
101 *elem_type = RankedTensorType::get(elem_shape, buffer_type.getElementType()); in GetSplitElementTypeAndCount()
196 auto elem_shape = GetTensorArrayElementShape(ta, module); in HandleTensorArrayV3Op() local
197 if (!elem_shape) return ta.emitOpError("unknown element shape"); in HandleTensorArrayV3Op()
203 if (failed(cutil::CreateInitBufferValue(*elem_shape, ta.size(), ta, in HandleTensorArrayV3Op()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_alias_analysis_test.cc1019 Shape elem_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
1022 HloInstruction::CreateParameter(0, elem_shape, "param")); in TEST_F()
1024 HloInstruction::CreateUnary(elem_shape, HloOpcode::kNegate, param0)); in TEST_F()
1026 HloInstruction::CreateUnary(elem_shape, HloOpcode::kNegate, negate)); in TEST_F()
1043 Shape elem_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
1046 HloInstruction::CreateParameter(0, elem_shape, "param")); in TEST_F()
1048 HloInstruction::CreateUnary(elem_shape, HloOpcode::kNegate, param0)); in TEST_F()
1050 HloInstruction::CreateUnary(elem_shape, HloOpcode::kNegate, negate)); in TEST_F()
Dtuple_points_to_analysis_test.cc904 Shape elem_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
906 0, ShapeUtil::MakeTupleShape({elem_shape, elem_shape}), "tuple")); in TEST_F()
908 HloInstruction::CreateGetTupleElement(elem_shape, tuple, 0)); in TEST_F()
910 HloInstruction::CreateGetTupleElement(elem_shape, tuple, 1)); in TEST_F()
912 HloInstruction::CreateBinary(elem_shape, HloOpcode::kAdd, gte0, gte1)); in TEST_F()
Dhlo_cost_analysis_test.cc454 auto elem_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
455 auto p2 = Parameter(&builder, 0, elem_shape, "x0"); in TEST_F()
456 auto p3 = Parameter(&builder, 1, elem_shape, "x1"); in TEST_F()
457 auto p4 = Parameter(&builder, 2, elem_shape, "y0"); in TEST_F()
458 auto p5 = Parameter(&builder, 3, elem_shape, "y1"); in TEST_F()
Dwhile_loop_simplifier.cc1082 const auto& elem_shape = while_shape.tuple_shapes(i); in TryMergeInductionVariables() local
1088 elem_shape, HloOpcode::kAdd, add_gte(instr, i), in TryMergeInductionVariables()
1089 add_binary_op(elem_shape, HloOpcode::kMultiply, in TryMergeInductionVariables()
Dbuffer_assignment_test.cc1558 auto elem_shape = f32vec4_; in TEST_F() local
1559 auto tuple_shape = ShapeUtil::MakeTupleShape({elem_shape}); in TEST_F()
1563 HloInstruction::CreateParameter(0, elem_shape, "sub_param")); in TEST_F()
1570 HloInstruction::CreateParameter(0, elem_shape, "param")); in TEST_F()
1599 auto elem_shape = f32vec4_; in TEST_F() local
1600 auto tuple_shape = ShapeUtil::MakeTupleShape({elem_shape}); in TEST_F()
1623 HloInstruction::CreateParameter(0, elem_shape, "param")); in TEST_F()
Dhlo_dataflow_analysis_test.cc2095 Shape elem_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2097 0, ShapeUtil::MakeTupleShape({elem_shape, elem_shape}), "tuple")); in TEST_F()
2099 HloInstruction::CreateGetTupleElement(elem_shape, tuple, 0)); in TEST_F()
2101 HloInstruction::CreateGetTupleElement(elem_shape, tuple, 1)); in TEST_F()
2103 HloInstruction::CreateBinary(elem_shape, HloOpcode::kAdd, gte0, gte1)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc584 for (const Shape& elem_shape : shape.tuple_shapes()) { in HumanString() local
585 StrAppend(&text, prefix, HumanString(elem_shape)); in HumanString()
608 for (const Shape& elem_shape : shape.tuple_shapes()) { in HumanStringWithLayout() local
609 StrAppend(&text, prefix, HumanStringWithLayout(elem_shape)); in HumanStringWithLayout()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlower_static_tensor_list.cc659 auto elem_shape = rewriter->create<TF::SliceOp>( in CreateCondTrueBranch() local
663 loc, resize_op.output_handle().getType(), elem_shape, size_diff); in CreateCondTrueBranch()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.cc365 TF_ASSIGN_OR_RETURN(xla::Shape elem_shape, in BuildComputation()
367 elem_shapes.push_back(elem_shape); in BuildComputation()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2338 const Shape& elem_shape = in HandleCustomCall() local
2340 TF_RET_CHECK(!elem_shape.IsTuple()) << "Nested tuples not implemented"; in HandleCustomCall()
2343 llvm::Value* addr = EmitBufferPointer(slice, elem_shape); in HandleCustomCall()