| /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/compiler/tf2xla/kernels/ |
| D | batchtospace_op.cc | 25 DataType input_dtype, const TensorShape& input_tensor_shape, in BatchToSpace()
|
| D | spacetobatch_op.cc | 25 DataType input_dtype, const TensorShape& input_tensor_shape, in SpaceToBatch()
|
| D | batch_norm_op.cc | 253 DataType input_dtype = ctx->input_type(0); in Compile() local
|
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
| D | string_lookup_test.py | 93 input_dtype): argument
|
| D | integer_lookup_test.py | 99 input_dtype): argument
|
| D | index_lookup_test.py | 327 input_dtype): argument
|
| /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 | 275 input_dtype=None): argument
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | testing_utils.py | 95 input_dtype=None, argument
595 input_dtype=None, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_tensor.cc | 123 const auto& input_dtype = input_tensor.dtype(); in ConvertTensor() local
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | composite_tensor_support_test.py | 127 input_dtype=None, argument
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 184 DataType input_dtype = ctx.input_dtype(i); in ShapeTraceString() local
404 Status OpKernelContext::input_dtype(StringPiece name, DataType* dtype) const { in input_dtype() function in tensorflow::OpKernelContext
|
| D | op_kernel.h | 1578 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 | 724 DataType input_dtype = in MoveHeadOutsideCompilationToHost() local
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 2908 tensorflow::DataType input_dtype(tensorflow::PyTensor_DataType(input)); local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 6372 NodeDef GetArgMinMaxNodeDef(DataType input_dtype, DataType output_dtype) { in TEST_P()
|