| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/ |
| H A D | reduction_benchmark.cc | 43 ArrayRef<bool> dynamic_dims, in GetReductionIR()
63 llvm::ArrayRef<bool> dynamic_dims, in GetSumF32IR()
70 llvm::ArrayRef<bool> dynamic_dims, in GetMeanF32IR()
|
| H A D | cwise_op_fusion_benchmark.cc | 20 static const char* dynamic_dims = R"( variable
|
| H A D | softmax_op_benchmark.cc | 34 std::string Softmax(llvm::ArrayRef<bool> dynamic_dims, in Softmax()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| H A D | ir_array.cc | 79 absl::Span<llvm::Value*> dynamic_dims, in Delinearize()
147 absl::Span<llvm::Value*> dynamic_dims, in Index()
449 const std::vector<llvm::Value*>& dynamic_dims, in Linearize()
|
| H A D | loop_emitter.cc | 38 std::vector<llvm::Value*> dynamic_dims, in LoopEmitter()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | tensor_list_utils.cc | 273 const std::vector<std::vector<xla::XlaOp>>& dynamic_dims, in CreateZerosTensorListWithShape()
335 std::vector<xla::XlaOp> dynamic_dims; in GetInitializedTensorListForElement() local
|
| H A D | broadcast_to_op.cc | 40 std::vector<bool> dynamic_dims; in Compile() local
|
| H A D | fill_op.cc | 52 std::vector<bool> dynamic_dims; in Compile() local
|
| H A D | sparse_to_dense_op.cc | 77 std::vector<bool> dynamic_dims; in Compile() local
|
| H A D | shape_op.cc | 393 std::vector<xla::XlaOp> dynamic_dims; in Compile() local
|
| H A D | while_op.cc | 529 std::vector<xla::XlaOp> dynamic_dims; in Compile() local
|
| H A D | tensor_list_ops.cc | 66 std::vector<xla::XlaOp> dynamic_dims; in GetTensorListDynamicDims() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| H A D | c_api_conversions.cc | 257 absl::Span<const bool> dynamic_dims = MakeSpan(c_shape->dynamic_dimensions); in FromC() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | pjrt_c_api_client.cc | 655 absl::Span<const bool> dynamic_dims = in set_shape() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_dynamic_shapes.py | 164 def create_symbolic_tensor(name, arg, shape_env, source=None, dynamic_dims=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | ir_emitter.cc | 2272 std::vector<llvm::Value*> dynamic_dims; in HandleSliceToDynamic() local
2321 std::vector<llvm::Value*> dynamic_dims; in HandlePadToStatic() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | ir_emitter_unnested.cc | 767 std::vector<llvm::Value*> dynamic_dims; in EmitPadToStatic() local
894 std::vector<llvm::Value*> dynamic_dims; in EmitSliceToDynamic() local
|