Home
last modified time | relevance | path

Searched refs:SymbolicShapesResolver (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/runtime/
H A Dsymbolic_shape.cc40 using SymbolicShape = SymbolicShapesResolver::SymbolicShape;
41 using StaticShape = SymbolicShapesResolver::StaticShape;
43 SymbolicShapesResolver::SymbolicShapesResolver( in SymbolicShapesResolver() function in xla::runtime::SymbolicShapesResolver
117 ArgumentConstraint SymbolicShapesResolver::constraint(size_t index) const { in constraint()
121 size_t SymbolicShapesResolver::num_arguments() const { in num_arguments()
125 bool SymbolicShapesResolver::has_argument_sizes(size_t index) const { in has_argument_sizes()
129 const StaticShape& SymbolicShapesResolver::argument_sizes(size_t index) const { in argument_sizes()
133 bool SymbolicShapesResolver::seen_static_size(size_t dim) const { in seen_static_size()
139 const SymbolicShapesResolver& resolver, ArgumentsRef arguments, in ResolveImpl()
221 llvm::ErrorOr<llvm::SmallVector<SymbolicShape>> SymbolicShapesResolver::Resolve( in Resolve()
[all …]
H A Dsymbolic_shape_test.cc35 using SymbolicShape = SymbolicShapesResolver::SymbolicShape;
93 SymbolicShapesResolver resolver(type, constraints); in TEST()
155 SymbolicShapesResolver resolver(type, constraints); in TEST()
217 SymbolicShapesResolver resolver(type, constraints); in TEST()
289 SymbolicShapesResolver resolver(type, constraints); in TEST()
315 SymbolicShapesResolver resolver(type, constraints); in TEST()
354 SymbolicShapesResolver resolver(type, constraints); in TEST()
375 SymbolicShapesResolver resolver(type, constraints); in TEST()
411 SymbolicShapesResolver resolver(type, constraints); in TEST()
454 SymbolicShapesResolver& resolver, ArrayRef<MemrefDesc> operands) { in Run()
[all …]
H A Dsymbolic_shape.h58 class SymbolicShapesResolver {
65 SymbolicShapesResolver(const FunctionType& signature,
H A Djit_executable.h175 SymbolicShapesResolver symbolic_shapes_resolver_;
H A Djit_executable.cc293 ErrorOr<llvm::SmallVector<SymbolicShapesResolver::SymbolicShape>> in GetExecutable()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir/transforms/runtime/
H A Dspecialization.cc47 using SymbolicShape = SymbolicShapesResolver::SymbolicShape;
84 auto shape = SymbolicShapesResolver::Normalize(symbolic_shape); in SpecializeOperandType()
H A Dspecialization.h62 llvm::ArrayRef<SymbolicShapesResolver::SymbolicShape> symbolic_shapes,
H A Djit_compiler.h50 using SymbolicShape = SymbolicShapesResolver::SymbolicShape;