Searched refs:UnalignedConstTensor (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_eigen_output_kernels.h | 215 typename TTypes<T>::UnalignedConstTensor bias(bias_base, num_rows); in operator()
242 typename TTypes<T>::UnalignedConstTensor bias(bias_base, num_rows);
278 typename TTypes<T>::UnalignedConstTensor scaling_factor(scaling_factor_base,
280 typename TTypes<T>::UnalignedConstTensor offset(offset_base, num_rows);
281 typename TTypes<T>::UnalignedConstTensor mean(mean_base, num_rows);
321 typename TTypes<T>::UnalignedConstTensor scaling_factor(scaling_factor_base,
323 typename TTypes<T>::UnalignedConstTensor offset(offset_base, num_rows);
324 typename TTypes<T>::UnalignedConstTensor mean(mean_base, num_rows);
|
| D | quantized_concat_op.cc | 49 typename TTypes<T, 1>::UnalignedConstTensor input_array(src, dims); in Copy()
|
| D | training_ops.cc | 784 auto g = typename TTypes<T>::UnalignedConstTensor(g_ptr + begin, t_size); in operator ()()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_types.h | 39 UnalignedConstTensor; typedef
|
| D | tensor.h | 572 typename TTypes<T, NDIMS>::UnalignedConstTensor unaligned_shaped(
886 typename TTypes<T, NDIMS>::UnalignedConstTensor Tensor::unaligned_shaped( in unaligned_shaped()
891 return typename TTypes<T, NDIMS>::UnalignedConstTensor(base<T>(), dims); in unaligned_shaped()
|
| D | tensor_test.cc | 452 TestReshape<TTypes<float, N>::UnalignedConstTensor, \ in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | decode_image_op.cc | 410 TTypes<uint16, 3>::UnalignedConstTensor buf(buffer.get(), height, width, in DecodePngV2()
635 TTypes<uint8, 3>::UnalignedConstTensor buf(buffer.get(), height, width, in DecodeBmpV2()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_concat_op.cc | 71 typename TTypes<T, 1>::UnalignedConstTensor input_array(src, dims); in Copy()
|