Searched defs:input_dtype (Results 1 – 10 of 10) sorted by relevance
28 const DataType input_dtype = ctx->input_dtype(0); in Compute() local
25 DataType input_dtype, const TensorShape& input_tensor_shape, in BatchToSpace()
25 DataType input_dtype, const TensorShape& input_tensor_shape, in SpaceToBatch()
127 DataType input_dtype = ctx->input_type(0); in Compile() local
38 DataType input_dtype = input->output_dtypes()[0]; in MakeDataset() local
70 def layer_test(layer_cls, kwargs=None, input_shape=None, input_dtype=None, argument
414 def input_dtype(self): member in DataFeeder
1499 inline DataType OpKernelContext::input_dtype(int index) const { in input_dtype() function
369 Status OpKernelContext::input_dtype(StringPiece name, DataType* dtype) const { in input_dtype() function in tensorflow::OpKernelContext
2185 TF_DataType input_dtype = TFE_TensorHandleDataType(handle->get()); local