Home
last modified time | relevance | path

Searched refs:bit_casted_shaped (Results 1 – 4 of 4) 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()
Ddecode_proto_op.cc895 data = tensor->bit_casted_shaped<uint8, 1>(flatshape).data(); in AccumulateFields()
/external/tensorflow/tensorflow/core/framework/
Dtensor_test.cc424 &Tensor::bit_casted_shaped<float, N>>({__VA_ARGS__}); \ in TEST_F()
426 &Tensor::bit_casted_shaped<float, N>>({__VA_ARGS__}); \ in TEST_F()
428 &Tensor::bit_casted_shaped<int32, N>>({__VA_ARGS__}); \ in TEST_F()
430 &Tensor::bit_casted_shaped<int32, N>>({__VA_ARGS__}); \ in TEST_F()
445 &Tensor::bit_casted_shaped<uint8, N>>({__VA_ARGS__}); \ in TEST_F()
457 &Tensor::bit_casted_shaped<int16, N>>({__VA_ARGS__}); \ in TEST_F()
477 EXPECT_DEATH((t.bit_casted_shaped<float, 0>({})), "4 vs. 480"); in TEST_F()
478 EXPECT_DEATH((t.bit_casted_shaped<float, 1>({119})), "476 vs. 480"); in TEST_F()
479 EXPECT_DEATH((t.bit_casted_shaped<float, 4>({2, 3, 4, 6})), "576 vs. 480"); in TEST_F()
483 EXPECT_DEATH((string_tensor.bit_casted_shaped<string, 1>({9})), "9 vs. 10"); in TEST_F()
Dtensor.h438 typename TTypes<T, NDIMS>::Tensor bit_casted_shaped(
505 typename TTypes<T, NDIMS>::ConstTensor bit_casted_shaped(
821 typename TTypes<T, NDIMS>::Tensor Tensor::bit_casted_shaped( in bit_casted_shaped() function
849 typename TTypes<T, NDIMS>::ConstTensor Tensor::bit_casted_shaped( in bit_casted_shaped() function