Searched defs:dynamic_dims (Results 1 – 11 of 11) sorted by relevance
38 std::vector<bool> dynamic_dims; in Compile() local
52 std::vector<int64> dynamic_dims; in Compile() local
271 const std::vector<std::vector<xla::XlaOp>>& dynamic_dims, in CreateZerosTensorListWithShape()333 std::vector<xla::XlaOp> dynamic_dims; in GetInitializedTensorListForElement() local
359 std::vector<xla::XlaOp> dynamic_dims; in Compile() local
516 std::vector<xla::XlaOp> dynamic_dims; in Compile() local
65 std::vector<xla::XlaOp> dynamic_dims; in GetTensorListDynamicDims() local
77 absl::Span<llvm::Value*> dynamic_dims, in Delinearize()116 absl::Span<llvm::Value*> dynamic_dims, in Index()420 const std::vector<llvm::Value*>& dynamic_dims, in Linearize()
39 std::vector<llvm::Value*> dynamic_dims, in LoopEmitter()
236 absl::Span<const bool> dynamic_dims = MakeSpan(c_shape->dynamic_dimensions); in FromC() local
2146 std::vector<llvm::Value*> dynamic_dims; in HandleSliceToDynamic() local2193 std::vector<llvm::Value*> dynamic_dims; in HandlePadToStatic() local
851 std::vector<llvm::Value*> dynamic_dims; in EmitPadToStaticFromMlir() local979 std::vector<llvm::Value*> dynamic_dims; in EmitSliceToDynamicFromMlir() local