| /external/tensorflow/tensorflow/core/kernels/ |
| D | guarantee_const_op.cc | 28 const DataType input_dtype = ctx->input_dtype(0); in Compute() local
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op_gpu_test.cc | 45 void MakeOp(DataType input_dtype = DT_FLOAT, in MakeOp()
236 void MakeOp(DataType input_dtype = DT_FLOAT, in MakeOp()
470 void MakeOp(DataType input_dtype = DT_FLOAT, in MakeOp()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batchtospace_op.cc | 25 DataType input_dtype, const TensorShape& input_tensor_shape, in BatchToSpace()
|
| D | spacetobatch_op.cc | 26 DataType input_dtype, const TensorShape& input_tensor_shape, in SpaceToBatch()
|
| D | batch_norm_op.cc | 257 DataType input_dtype = ctx->input_type(0); in Compile() local
|
| D | sharding_util_ops.cc | 140 const DataType input_dtype) { in CompileInternal()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | algorithm_selector.cc | 118 TacticID tactic, nvinfer1::DataType input_dtype, in AllowShuffleAlgorithm()
|
| D | convert_nodes_test.cc | 8439 NodeDef GetArgMinMaxNodeDef(DataType input_dtype, DataType output_dtype) { in GetArgMinMaxNodeDef()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | unique_dataset_op.cc | 219 DataType input_dtype = input->output_dtypes()[0]; in MakeDataset() local
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | sequential.py | 270 input_dtype=None): argument
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | testing_utils.py | 90 input_dtype=None, argument
572 input_dtype=None, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_tensor.cc | 118 const auto& input_dtype = input_tensor.dtype(); in ConvertTensor() local
|
| /external/tensorflow/tensorflow/core/ir/importexport/ |
| D | convert_tensor.cc | 115 const auto& input_dtype = input_tensor.dtype(); in ConvertTensor() local
|
| /external/tensorflow/tensorflow/core/tfrt/graph_executor/ |
| D | graph_executor.cc | 579 auto input_dtype = input_tensor_dtypes[i]; in GetOrCreateLoadedClientGraph() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 225 DataType input_dtype = ctx.input_dtype(i); in ShapeTraceString() local
417 Status OpKernelContext::input_dtype(StringPiece name, DataType* dtype) const { in input_dtype() function in tensorflow::OpKernelContext
|
| D | op_kernel.h | 1549 inline DataType OpKernelContext::input_dtype(int index) const { in input_dtype() function
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 694 DataType input_dtype = in MoveHeadOutsideCompilationToHost() local
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 2988 tensorflow::DataType input_dtype(tensorflow::PyTensor_DataType(input)); local
|