| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | broadcast_to_op.cc | 38 std::vector<bool> dynamic_dims; in Compile() local
40 context, context->ResolveInputDynamismIntoPredVector(1, &dynamic_dims)); in Compile()
41 for (int64 dim = 0; dim < dynamic_dims.size(); ++dim) { in Compile()
42 if (dynamic_dims[dim]) { in Compile()
|
| D | fill_op.cc | 52 std::vector<int64> dynamic_dims; in Compile() local
53 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector("dims", &dynamic_dims)); in Compile()
58 if (dynamic_dims[i] == -1) { in Compile()
|
| D | tensor_list_utils.cc | 271 const std::vector<std::vector<xla::XlaOp>>& dynamic_dims, in CreateZerosTensorListWithShape() argument
275 TF_RET_CHECK(dynamic_dims.size() == tuple_size - 1); in CreateZerosTensorListWithShape()
282 TF_RET_CHECK(dynamic_dims[i].size() == shape.dimensions_size()); in CreateZerosTensorListWithShape()
284 zeros = xla::SetDimensionSize(zeros, dynamic_dims[i][dim], dim); in CreateZerosTensorListWithShape()
333 std::vector<xla::XlaOp> dynamic_dims; in GetInitializedTensorListForElement() local
335 dynamic_dims.push_back(leading_dim_dynamic_size); in GetInitializedTensorListForElement()
343 dynamic_dims.push_back(xla::GetDimensionSize(sub_element, dim)); in GetInitializedTensorListForElement()
345 list_dynamic_dims.push_back(dynamic_dims); in GetInitializedTensorListForElement()
|
| D | tensor_list_ops.cc | 65 std::vector<xla::XlaOp> dynamic_dims; in GetTensorListDynamicDims() local
67 dynamic_dims.push_back(ctx->Input(1)); in GetTensorListDynamicDims()
69 dynamic_dims.push_back( in GetTensorListDynamicDims()
77 dynamic_dims.push_back(dynamic_dim_size); in GetTensorListDynamicDims()
79 dynamic_dims.push_back( in GetTensorListDynamicDims()
83 list_dynamic_dims.push_back(dynamic_dims); in GetTensorListDynamicDims()
|
| D | while_op.cc | 516 std::vector<xla::XlaOp> dynamic_dims; in Compile() local
525 dynamic_dims.push_back(leading_dim_size); in Compile()
528 dynamic_dims.push_back( in Compile()
540 dynamic_dims.push_back( in Compile()
543 list_dynamic_dims.push_back(dynamic_dims); in Compile()
|
| D | shape_op.cc | 359 std::vector<xla::XlaOp> dynamic_dims; in Compile() local
363 dynamic_dims.push_back(xla::GetDimensionSize(sub_element, dim)); in Compile()
365 list_dynamic_dims.push_back(dynamic_dims); in Compile()
|
| D | tensor_list_utils.h | 84 const std::vector<std::vector<xla::XlaOp>>& dynamic_dims, xla::XlaOp* list);
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.cc | 77 absl::Span<llvm::Value*> dynamic_dims, in Delinearize() argument
79 CHECK_EQ(shape.dimensions_size(), dynamic_dims.size()); in Delinearize()
93 b->CreateURem(quot, dynamic_dims[dimension], "dim_value"); in Delinearize()
94 divisor = b->CreateMul(divisor, dynamic_dims[dimension], "divisor"); in Delinearize()
116 absl::Span<llvm::Value*> dynamic_dims, in Index() argument
127 Delinearize(&multidim_, linear, shape, dynamic_dims, b); in Index()
420 const std::vector<llvm::Value*>& dynamic_dims, in Linearize() argument
424 CHECK_EQ(size(), dynamic_dims.size()); in Linearize()
434 multiplier = builder->CreateMul(multiplier, dynamic_dims[i], in Linearize()
|
| D | loop_emitter.cc | 39 std::vector<llvm::Value*> dynamic_dims, in LoopEmitter() argument
42 CHECK_EQ(dynamic_dims.size(), shape_.dimensions_size()); in LoopEmitter()
43 dynamic_dims_ = std::move(dynamic_dims); in LoopEmitter()
|
| D | ir_array.h | 72 absl::Span<llvm::Value*> dynamic_dims, llvm::IRBuilder<>* b);
161 llvm::Value* Linearize(const std::vector<llvm::Value*>& dynamic_dims,
187 const Shape& shape, absl::Span<llvm::Value*> dynamic_dims,
|
| D | loop_emitter.h | 50 std::vector<llvm::Value*> dynamic_dims, llvm::IRBuilder<>* b);
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | c_api_conversions.cc | 236 absl::Span<const bool> dynamic_dims = MakeSpan(c_shape->dynamic_dimensions); in FromC() local
245 dims, dynamic_dims, std::move(tuple_shapes)); in FromC()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2146 std::vector<llvm::Value*> dynamic_dims; in HandleSliceToDynamic() local
2161 dynamic_dims.push_back(b_.CreateIntCast(dyn_dim_size, b_.getInt64Ty(), in HandleSliceToDynamic()
2176 llvm::Value* linear_index = array_index.Linearize(dynamic_dims, &b_); in HandleSliceToDynamic()
2184 dynamic_dims, &b_) in HandleSliceToDynamic()
2193 std::vector<llvm::Value*> dynamic_dims; in HandlePadToStatic() local
2227 dynamic_dims.push_back(b_.CreateIntCast(dyn_dim_size, b_.getInt64Ty(), in HandlePadToStatic()
2240 llvm::Value* linear_index = array_index.Linearize(dynamic_dims, &b_); in HandlePadToStatic()
2248 llvm_ir::LoopEmitter(loop_body_emitter, input_shape, dynamic_dims, &b_) in HandlePadToStatic()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 851 std::vector<llvm::Value*> dynamic_dims; in EmitPadToStaticFromMlir() local
864 dynamic_dims.push_back(dyn_dim_size); in EmitPadToStaticFromMlir()
880 b_.CreateStore(dynamic_dims[i - 1], in EmitPadToStaticFromMlir()
890 for (llvm::Value* dynamic_dim : dynamic_dims) { in EmitPadToStaticFromMlir()
916 absl::MakeSpan(dynamic_dims), &b_); in EmitPadToStaticFromMlir()
979 std::vector<llvm::Value*> dynamic_dims; in EmitSliceToDynamicFromMlir() local
984 dynamic_dims.push_back(dyn_dim_size); in EmitSliceToDynamicFromMlir()
1002 dynamic_dims[dim_index], in EmitSliceToDynamicFromMlir()
1011 for (llvm::Value* dynamic_dim : dynamic_dims) { in EmitSliceToDynamicFromMlir()
1037 absl::MakeSpan(dynamic_dims), &b_); in EmitSliceToDynamicFromMlir()
|