Searched refs:int32_input (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_pooling_ops.cc | 83 Tensor int32_input(DT_INT32, tensor_in.shape()); in Compute() local
84 int32_input.flat<int32>() = tensor_in.flat<T>().template cast<int32>(); in Compute()
85 SpatialAvgPool<Device, int32>(context, &int32_output, int32_input, params, in Compute()
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | bitcast_op_test.cc | 91 Tensor int32_input(DT_UINT32, {4, 6}); in TEST() local
92 TestBitcastOp(&int32_input, DT_UINT8, TensorShape({4, 6, 4}), error::OK); in TEST()
|