Searched defs:dynamic_dims (Results 1 – 16 of 16) sorted by relevance
43 ArrayRef<bool> dynamic_dims, in GetReductionIR()63 llvm::ArrayRef<bool> dynamic_dims, in GetSumF32IR()70 llvm::ArrayRef<bool> dynamic_dims, in GetMeanF32IR()
20 static const char* dynamic_dims = R"( variable
34 std::string Softmax(llvm::ArrayRef<bool> dynamic_dims, in Softmax()
40 std::vector<bool> dynamic_dims; in Compile() local
52 std::vector<bool> dynamic_dims; in Compile() local
77 std::vector<bool> dynamic_dims; in Compile() local
273 const std::vector<std::vector<xla::XlaOp>>& dynamic_dims, in CreateZerosTensorListWithShape()335 std::vector<xla::XlaOp> dynamic_dims; in GetInitializedTensorListForElement() local
393 std::vector<xla::XlaOp> dynamic_dims; in Compile() local
529 std::vector<xla::XlaOp> dynamic_dims; in Compile() local
66 std::vector<xla::XlaOp> dynamic_dims; in GetTensorListDynamicDims() local
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()
38 std::vector<llvm::Value*> dynamic_dims, in LoopEmitter()
257 absl::Span<const bool> dynamic_dims = MakeSpan(c_shape->dynamic_dimensions); in FromC() local
655 absl::Span<const bool> dynamic_dims = in set_shape() local
2272 std::vector<llvm::Value*> dynamic_dims; in HandleSliceToDynamic() local2321 std::vector<llvm::Value*> dynamic_dims; in HandlePadToStatic() local
767 std::vector<llvm::Value*> dynamic_dims; in EmitPadToStatic() local894 std::vector<llvm::Value*> dynamic_dims; in EmitSliceToDynamic() local