| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_xent_op.h | 34 return To32Bit(typename TTypes<T>::ConstVec(in.data(), in.dimensions())); in To32BitConst()
40 return To32Bit(typename TTypes<T>::ConstMatrix(in.data(), in.dimensions())); in To32BitConst()
148 To32Bit(maximum).device(d) = To32Bit(logits).maximum(along_row); in Compute()
217 To32Bit(backprop).device(d) = in Compute()
218 To32Bit(logits) - in Compute()
219 To32Bit(scratch).reshape(batch_by_one).broadcast(one_by_class); in Compute()
222 To32Bit(scratch).device(d) = To32Bit(backprop).exp().sum(along_class); in Compute()
229 sparse_xent_helpers::To32BitConst<T>(scratch), To32Bit(labels), in Compute()
231 To32Bit(loss).device(d) = in Compute()
232 To32Bit(backprop).generate(sparse_xent_loss_gen).sum(along_class); in Compute()
[all …]
|
| D | cwise_ops_gpu_common.cu.h | 44 To32Bit(out).device(d) = To32Bit(in).unaryExpr(typename Functor::func());
54 To32Bit(out).device(d) =
55 To32Bit(in0).binaryExpr(in1, typename Functor::func());
65 To32Bit(out).device(d) = To32Bit(in).unaryExpr(Unary(scalar.data()));
75 To32Bit(out).device(d) = To32Bit(in).unaryExpr(Unary(scalar.data()));
92 To32Bit(out).device(d) =
93 To32Bit(in0).binaryExpr(To32Bit(in1).broadcast(bcast1), func);
97 To32Bit(out).device(d) =
98 To32Bit(in0).broadcast(bcast0).binaryExpr(To32Bit(in1), func);
102 To32Bit(out).device(d) = To32Bit(in0).broadcast(bcast0).binaryExpr(
[all …]
|
| D | constant_op_gpu.cu.cc | 75 To32Bit(out).device(d) = To32Bit(out).nullaryExpr(f); in operator ()()
88 To32Bit(out).device(d) = To32Bit(out).constant(T(0)); in operator ()()
101 To32Bit(out).device(d) = To32Bit(out).constant(T(1)); in operator ()()
|
| D | fill_functor.cu.cc | 75 To32Bit(out).device(d) = To32Bit(out).nullaryExpr(f); in operator ()()
88 To32Bit(out).device(d) = To32Bit(out).constant(T(0)); in operator ()()
108 To32Bit(out).device(d) = To32Bit(out).constant(T(1)); in operator ()()
|
| D | cwise_op_gpu_select.cu.cc | 48 To32Bit(out).device(d) = in operator ()()
49 To32Bit(cond_flat).select(To32Bit(then_flat), To32Bit(else_flat)); in operator ()()
67 To32Bit(out).device(d) = cond.reshape(rank1) in operator ()()
|
| D | bias_op.h | 45 To32Bit(output).reshape(one_d).device(d) = in operator()
46 To32Bit(input).reshape(one_d) + To32Bit(bias).broadcast(bcast); in operator()
|
| D | strided_slice_op.h | 46 To32Bit(output).device(d) = in operator()
47 To32Bit(input).stridedSlice(start_i, stop_i, strides_i); in operator()
95 To32Bit(output).stridedSlice(start_i, stop_i, strides_i).device(d) =
120 To32Bit(output).stridedSlice(start_i, stop_i, strides_i).device(d) =
121 To32Bit(input);
|
| D | slice_op.h | 44 To32Bit(output).device(d) = To32Bit(input).slice(indices, sizes); in operator()
|
| D | avgpooling_op.h | 35 To32Bit(output).swap_layout().device(d) = Eigen::SpatialAvgPooling( in operator()
36 To32Bit(input).swap_layout(), window_cols, window_rows, col_stride, in operator()
|
| D | pad_op.h | 38 To32Bit(output).device(d) = To32Bit(input).pad(paddings, pad_value); in operator()
|
| D | multinomial_op_gpu.cu.cc | 98 To32Bit(scores).device(d) = in operator ()()
99 To32Bit(logits).reshape(boc).broadcast(oso).template cast<float>() - in operator ()()
100 ((-((To32Bit(noises) + 2e-30f).log())).log()); in operator ()()
|
| D | cwise_ops_gpu_gradients.cu.h | 46 To32Bit(out).device(d) =
47 To32Bit(in1).binaryExpr(in2, typename Functor::func());
|
| D | split_lib_gpu.cu.cc | 38 To32Bit(output).device(d) = To32Bit(input).slice(slice_indices, slice_sizes); in operator ()()
47 To32Bit(output).device(d) = To32Bit(input).slice(slice_indices, slice_sizes); in operator ()()
|
| D | conv_2d.h | 130 To32Bit(input_backward).device(d) = Eigen::SpatialConvolutionBackwardInput(
131 To32Bit(filter), To32Bit(output_backward), input_backward.dimension(2),
175 To32Bit(input_backward).device(d) =
177 To32Bit(filter), To32Bit(output_backward), padded_cols, padded_rows,
|
| D | broadcast_to_op.h | 38 To32Bit(out).device(device) = To32Bit(in).broadcast(bcast); in DoBCast32Bit()
|
| D | tile_functor.h | 53 To32Bit(y).device(d) = To32Bit(x).broadcast(b); in TileUsingEigen()
|
| D | topk_op_gpu.h | 464 auto input_indices_t = To32Bit(input_indices.flat<int32>());
543 To32Bit(indices).device(d) =
544 To32Bit(temp_indices.matrix<int32>()).slice(slice_indices, slice_sizes);
545 To32Bit(values).device(d) =
546 To32Bit(temp_values.matrix<T>()).slice(slice_indices, slice_sizes);
|
| D | scan_ops.h | 39 To32Bit(out).device(d) = in operator()
40 To32Bit(in).reverse(dims).scan(1, reducer, exclusive).reverse(dims); in operator()
|
| D | pooling_ops_common.cc | 341 To32Bit(const_transformed_input.tensor<T, 4>()), in Compute()
344 To32Bit(padded_input.tensor<T, 4>()), data_format)); in Compute()
648 To32Bit(const_transformed_input.tensor<T, 4>()), in Compute()
651 To32Bit(padded_input.tensor<T, 4>()), in Compute()
737 To32Bit(const_cast<const Tensor&>(transformed_and_padded_input_backprop) in Compute()
742 To32Bit(transformed_input_backprop.template tensor<T, 4>()), in Compute()
|
| D | concat_lib_gpu_impl.cu.cc | 127 To32Bit(*output).slice(offset, size).device(gpu_device) = in ConcatGPUSlice()
128 To32Bit(*inputs_flat[i]); in ConcatGPUSlice()
|
| D | conv_grad_input_ops.cc | 297 To32Bit(filter.tensor<T, 4>()), in operator ()()
298 To32Bit(transformed_filter.tensor<T, 4>())); in operator ()()
529 To32Bit(const_cast<const Tensor&>(pre_transformed_in_backprop) in operator ()()
534 To32Bit(in_backprop_remove_padding.tensor<T, 4>()), compute_data_format, in operator ()()
|
| D | reduction_gpu_kernels.cu.h | 1134 FillIdentityEigenImpl(d, To32Bit(out), reducer);
1159 FillIdentityEigenImpl(d, To32Bit(out), reducer);
1195 FillIdentityEigenImpl(d, To32Bit(out), reducer);
1238 FillIdentityEigenImpl(d, To32Bit(out), reducer);
1261 FillIdentityEigenImpl(d, To32Bit(out), reducer);
1284 FillIdentityEigenImpl(d, To32Bit(out), reducer);
1304 FillIdentityEigenImpl(d, To32Bit(out), reducer);
1324 FillIdentityEigenImpl(d, To32Bit(out), reducer);
1343 FillIdentityEigenImpl(d, To32Bit(out), reducer);
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | extract_image_patches_op.h | 37 auto output_32bit = To32Bit(output); in operator()
39 To32Bit(input) in operator()
|
| D | extract_volume_patches_op.h | 37 auto output_32bit = To32Bit(output); in operator()
39 To32Bit(input) in operator()
|
| D | mirror_pad_op.cc | 125 context->eigen_device<Device>(), To32Bit(output->tensor<T, i>()), \ in Compute()
126 To32Bit(in0.tensor<T, i>()), paddings, offset_); \ in Compute()
336 context->eigen_device<Device>(), To32Bit(output->tensor<T, k>()), \ in Compute()
337 To32Bit(in0.tensor<T, k>()), paddings, offset_, \ in Compute()
338 To32Bit(scratch.tensor<T, k>())); \ in Compute()
|