Home
last modified time | relevance | path

Searched refs:bit_casted_shaped (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op_impl.h97 result->bit_casted_shaped<Proxy, NDIM>(processing_dims), in HandleStridedSliceCase()
110 result->bit_casted_shaped<Proxy, NDIM>(processing_dims), in HandleStridedSliceCase()
137 context->input(4).bit_casted_shaped<Proxy, NDIM>(processing_dims), in HandleStridedSliceGradCase()
158 context->input(4).bit_casted_shaped<Proxy, NDIM>(processing_dims), in operator()
177 result->bit_casted_shaped<Proxy, 1>(processing_dims), in operator()
178 context->input(4).bit_casted_shaped<Proxy, 1>(processing_dims)); in operator()
Dfingerprint_op.cc128 auto data = input.bit_casted_shaped<uint8, 2>( in Compute()
Dfingerprint_op_test.cc124 auto buffer = tensor.bit_casted_shaped<uint8, 1>({size}); in TEST_F()
Ddequantize_op.cc122 auto input_tensor = input.template bit_casted_shaped<T, 3>( in Compute()
Dquantize_op.cc153 auto output_tensor = output->template bit_casted_shaped<T, 3>( in Compute()
Ddecode_proto_op.cc898 data = tensor->bit_casted_shaped<uint8, 1>(flatshape).data(); in AccumulateFields()
/external/tensorflow/tensorflow/core/framework/
Dtensor_test.cc455 &Tensor::bit_casted_shaped<float, N>>({__VA_ARGS__}); \ in TEST_F()
457 &Tensor::bit_casted_shaped<float, N>>({__VA_ARGS__}); \ in TEST_F()
459 &Tensor::bit_casted_shaped<int32, N>>({__VA_ARGS__}); \ in TEST_F()
461 &Tensor::bit_casted_shaped<int32, N>>({__VA_ARGS__}); \ in TEST_F()
477 &Tensor::bit_casted_shaped<uint8, N>>({__VA_ARGS__}); \ in TEST_F()
490 &Tensor::bit_casted_shaped<int16, N>>({__VA_ARGS__}); \ in TEST_F()
510 EXPECT_DEATH((t.bit_casted_shaped<float, 0>({})), "4 vs. 480"); in TEST_F()
511 EXPECT_DEATH((t.bit_casted_shaped<float, 1>({119})), "476 vs. 480"); in TEST_F()
512 EXPECT_DEATH((t.bit_casted_shaped<float, 4>({2, 3, 4, 6})), "576 vs. 480"); in TEST_F()
516 EXPECT_DEATH((string_tensor.bit_casted_shaped<tstring, 1>({9})), "9 vs. 10"); in TEST_F()
Dtensor.h501 typename TTypes<T, NDIMS>::Tensor bit_casted_shaped(
568 typename TTypes<T, NDIMS>::ConstTensor bit_casted_shaped(
849 typename TTypes<T, NDIMS>::Tensor Tensor::bit_casted_shaped( in bit_casted_shaped() function
877 typename TTypes<T, NDIMS>::ConstTensor Tensor::bit_casted_shaped( in bit_casted_shaped() function