| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | compile_mlir_util.cc | 84 mlir::ModuleOp module, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in GetXlaInputShapes()
230 Status RefineShapes(llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in RefineShapes()
427 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in CompileMlirSetup()
444 llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in BuildHloFromTf()
467 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in PopulateResultIOInfo()
488 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in CompileMlirToXlaHlo()
517 llvm::StringRef mlir_module_string, llvm::ArrayRef<TensorShape> arg_shapes, in CompileSerializedMlirToXlaHlo()
618 llvm::SmallVector<TensorOrResourceShape, 4>& arg_shapes) { in CompileGraphSetup()
660 llvm::SmallVector<TensorOrResourceShape, 4> arg_shapes; in BuildHloFromModule() local
676 llvm::SmallVector<TensorOrResourceShape, 4> arg_shapes; in CompileGraphToXlaHlo() local
|
| D | tf_xla_mlir_translate.cc | 119 llvm::SmallVectorImpl<TensorOrResourceShape>& arg_shapes) { in ParseArgumentShapes()
241 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in CompileMlirToXlaHloViaBuilder()
297 llvm::SmallVector<TensorOrResourceShape, 4> arg_shapes; in MlirTfToHloTextTranslateFunctionImpl() local
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_compile.cc | 163 const std::vector<PartialTensorShape>& arg_shapes, in OptimizeGraph()
249 const std::vector<TensorShape>& arg_shapes, in BuildComputationArgumentDescriptions()
336 const std::vector<PartialTensorShape>& arg_shapes, Graph* graph, in RunShapeInferenceOnComputation()
370 const std::vector<TensorShape>& arg_shapes, in CompileTFFunctionToHlo()
451 absl::Span<const TensorShape> arg_shapes, in GetShardingInfo()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile.proto | 45 repeated TensorShapeProto arg_shapes = 6; field
|
| D | tpu_compile_op_support.cc | 168 std::vector<Shape> arg_shapes; in GetPerDeviceShape() local
345 const std::vector<TensorShape>& arg_shapes) { in CreateTpuCompilationRequest()
449 std::vector<TensorShape>* arg_shapes) { in ComputeArgumentShapes()
|
| D | tpu_compile_op_impl.cc | 31 const std::vector<TensorShape>& arg_shapes, in Compile()
|
| D | tpu_compile_op_common.cc | 152 std::vector<TensorShape> arg_shapes; in CompileLocallyAndFillHostCache() local
424 const std::vector<TensorShape>& arg_shapes, in RegisterXLAFingerprints()
|
| D | tpu_functional_ops.cc | 386 std::map<int, InferredShape>& arg_shapes, EdgeShapes& tpu_input_shapes, in GetInputOutputInfo()
515 std::vector<TensorShape> arg_shapes; in MaybeRegisterFingerprint() local
1912 std::map<int, InferredShape>& arg_shapes, in InferShapesWithResourceVar()
2108 std::map<int, InferredShape> arg_shapes; in OptimizeTpuInputOutputTensors() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | shape_inference.cc | 46 const std::map<int, InferredShape>& arg_shapes, in PropagateShapes()
246 Status InferShapes(Graph* graph, const std::map<int, InferredShape>& arg_shapes, in InferShapes()
|
| D | xla_compilation_cache.h | 118 arg_shapes; member
|
| D | encapsulate_util.cc | 325 std::map<int, InferredShape> arg_shapes; in PerformStaticShapeInferenceBeforeEncapsulation() local
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op.cc | 48 const std::vector<TensorShape> arg_shapes; member
279 std::vector<TensorShape> arg_shapes; in SetupArgs() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_jit_compiled_cpu_function.cc | 111 std::vector<const xla::Shape*> arg_shapes; in Compile() local
|
| D | xla_compiler.cc | 1033 std::vector<xla::Shape> arg_shapes(input_to_args->size()); in BuildArguments() local
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | utils.cc | 76 std::vector<Shape> arg_shapes; in GetShardedProgramShapes() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 372 absl::Span<const Shape* const> arg_shapes, const int64_t dimension) { in InferConcatOpShape()
1106 absl::Span<const Shape* const> arg_shapes, const ProgramShape& to_apply, in InferMapShape()
2228 absl::Span<const Shape* const> arg_shapes, in InferReduceShape()
3316 absl::Span<const Shape* const> arg_shapes, const ProgramShape& to_apply) { in InferCallShape()
|
| D | hlo_parser.cc | 1531 absl::InlinedVector<const Shape*, 2> arg_shapes; in ParseInstructionRhs() local
1552 absl::InlinedVector<const Shape*, 2> arg_shapes; in ParseInstructionRhs() local
1692 absl::InlinedVector<const Shape*, 2> arg_shapes; in ParseInstructionRhs() local
1907 absl::InlinedVector<const Shape*, 2> arg_shapes; in ParseInstructionRhs() local
1934 absl::InlinedVector<const Shape*, 2> arg_shapes; in ParseInstructionRhs() local
1968 absl::InlinedVector<const Shape*, 2> arg_shapes; in ParseInstructionRhs() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | fake_quantize_ops.cc | 102 std::vector<xla::Shape> arg_shapes; in BuildFakeQuantCustomCall() local
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 1966 const ParameterInfo& params_info, std::vector<InferredShape>* arg_shapes, in GetArgAndRetvalShapes()
2186 const std::vector<InferredShape>& arg_shapes, in AssignArgsAndRetvalsToCores()
2516 const Node& replicate_node, const std::vector<InferredShape>& arg_shapes, in BuildDynamicShapeNodes()
2648 const std::vector<InferredShape>& arg_shapes, in BuildCompileNode()
3216 const std::vector<InferredShape>& arg_shapes, in CreateOrGetPerHostVariableCopy()
3303 const std::vector<InferredShape>& arg_shapes, in BuildExecuteNodes()
4811 std::vector<InferredShape> arg_shapes, retval_shapes; in RewriteTPUReplicateNode() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 1639 llvm::SmallVectorImpl<xla::Shape>* arg_shapes, in SetEntryTupleShapesAndLeafReplication()
1677 llvm::SmallVectorImpl<xla::Shape>* arg_shapes) { in SetEntryTupleShardings()
1714 llvm::SmallVector<xla::Shape, 4> arg_shapes; in LowerBasicBlockAsFunction() local
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | function.py | 915 arg_shapes=None, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | shape_inference.cc | 1822 ArrayRef<ArrayRef<int64_t>> arg_shapes, in InferShapeForFunction()
|