Searched refs:tensor_input (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | pooling_ops_test.py | 768 tensor_input = np.random.rand(*input_shape).astype(dtype)
770 t = constant_op.constant(tensor_input, shape=input_shape)
774 t = constant_op.constant(tensor_input, shape=input_shape)
784 tensor_input = np.random.random_integers(0, 3, input_shape).astype(dtype)
787 t = constant_op.constant(tensor_input, shape=input_shape)
796 t = constant_op.constant(tensor_input, shape=input_shape)
814 tensor_input = np.random.random_integers(0, 3, input_shape).astype(dtype)
816 t = constant_op.constant(tensor_input, shape=input_shape)
819 grad_in = constant_op.constant(tensor_input, shape=input_shape)
825 t = constant_op.constant(tensor_input, shape=input_shape)
[all …]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | importer_test.py | 1071 tensor_input = np.ones(input_shape, dtype=np.float32)
1072 t = constant_op.constant(tensor_input, shape=input_shape)
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tensor_tracer.py | 1121 input_ops = [tensor_input.op for tensor_input in head_op.inputs]
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 3785 TRT_TensorOrWeights tensor_input, in ConvertMatMulHelper() argument
3789 if (!tensor_input.is_tensor()) { in ConvertMatMulHelper()
3792 const nvinfer1::ITensor* tensor = tensor_input.tensor(); in ConvertMatMulHelper()
3810 tensor_input, input_dim, /*validation_only=*/false, &tensor)); in ConvertMatMulHelper()
|