| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | compile_mlir_util.cc | 83 mlir::ModuleOp module, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in GetXlaInputShapes()
228 Status RefineShapes(llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in RefineShapes()
407 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in CompileMlirSetup()
424 llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in BuildHloFromTf()
447 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in PopulateResultIOInfo()
468 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in CompileMlirToXlaHlo()
491 llvm::StringRef mlir_module_string, llvm::ArrayRef<TensorShape> arg_shapes, in CompileSerializedMlirToXlaHlo()
590 llvm::SmallVector<TensorOrResourceShape, 4>& arg_shapes) { in CompileGraphSetup()
624 llvm::SmallVector<TensorOrResourceShape, 4> arg_shapes; in BuildHloFromModule() local
639 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/kernels/ |
| D | tpu_compile_op_common.cc | 180 const std::vector<TensorShape>& arg_shapes, in BuildComputationArgumentDescriptions()
256 absl::Span<const TensorShape> arg_shapes, in GetShardingInfo()
281 const std::vector<TensorShape>& arg_shapes, in CompileTFFunctionToHlo()
457 const std::vector<PartialTensorShape>& arg_shapes, Graph* graph, in RunShapeInferenceOnComputation()
489 const std::vector<PartialTensorShape>& arg_shapes, in OptimizeGraph()
598 std::vector<TensorShape> arg_shapes; in CompileLocallyAndFillHostCache() local
|
| D | tpu_compile.proto | 45 repeated TensorShapeProto arg_shapes = 6; field
|
| D | tpu_compile_op_impl.cc | 31 const std::vector<TensorShape>& arg_shapes, in Compile()
|
| D | tpu_compile_op_support.cc | 166 std::vector<Shape> arg_shapes; in GetPerDeviceShape() local
340 const std::vector<TensorShape>& arg_shapes) { in CreateTpuCompilationRequest()
443 std::vector<TensorShape>* arg_shapes) { in ComputeArgumentShapes()
|
| /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 | 104 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
277 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 | 1023 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 | 383 absl::Span<const Shape* const> arg_shapes, const int64 dimension) { in InferConcatOpShape()
1117 absl::Span<const Shape* const> arg_shapes, const ProgramShape& to_apply, in InferMapShape()
2114 absl::Span<const Shape* const> arg_shapes, in InferReduceShape()
3179 absl::Span<const Shape* const> arg_shapes, const ProgramShape& to_apply) { in InferCallShape()
|
| D | hlo_parser.cc | 1408 absl::InlinedVector<const Shape*, 2> arg_shapes; in ParseInstructionRhs() local
1555 absl::InlinedVector<const Shape*, 2> arg_shapes; in ParseInstructionRhs() local
1770 absl::InlinedVector<const Shape*, 2> arg_shapes; in ParseInstructionRhs() local
1797 absl::InlinedVector<const Shape*, 2> arg_shapes; in ParseInstructionRhs() local
1831 absl::InlinedVector<const Shape*, 2> arg_shapes; in ParseInstructionRhs() local
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 1694 const ParameterInfo& params_info, std::vector<InferredShape>* arg_shapes, in GetArgAndRetvalShapes()
1914 const std::vector<InferredShape>& arg_shapes, in AssignArgsAndRetvalsToCores()
2234 const Node& replicate_node, const std::vector<InferredShape>& arg_shapes, in BuildDynamicShapeNodes()
2307 const std::vector<InferredShape>& arg_shapes, in BuildCompileNode()
2796 const std::vector<InferredShape>& arg_shapes, in CreateOrGetPerHostVariableCopy()
2890 const std::vector<InferredShape>& arg_shapes, in BuildExecuteNodes()
4252 std::vector<InferredShape> arg_shapes, retval_shapes; in RewriteTPUReplicateNode() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | fake_quantize_ops.cc | 99 std::vector<xla::Shape> arg_shapes; in BuildFakeQuantCustomCall() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 1444 llvm::SmallVectorImpl<xla::Shape>* arg_shapes, in SetEntryTupleShapesAndLeafReplication()
1482 llvm::SmallVectorImpl<xla::Shape>* arg_shapes) { in SetEntryTupleShardings()
1519 llvm::SmallVector<xla::Shape, 4> arg_shapes; in LowerBasicBlockAsFunction() local
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | function.py | 912 arg_shapes=None, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | shape_inference.cc | 1594 ArrayRef<ArrayRef<int64_t>> arg_shapes, in InferShapeForFunction()
|