| /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 | cwise_ops_sycl_common.h | 44 To32Bit(out).device(d) = To32Bit(in).unaryExpr(typename Functor::func());
54 To32Bit(out).device(d) =
55 To32Bit(in0).binaryExpr(To32Bit(in1), typename Functor::func());
95 To32Bit(out).device(d) =
96 To32Bit(in0).binaryExpr(To32Bit(in1).broadcast(bcast1), func);
100 To32Bit(out).device(d) =
101 To32Bit(in0).broadcast(bcast0).binaryExpr(To32Bit(in1), func);
105 To32Bit(out).device(d) = To32Bit(in0).broadcast(bcast0).binaryExpr(
106 To32Bit(in1).broadcast(bcast1), func);
|
| D | fill_functor.cu.cc | 74 To32Bit(out).device(d) = To32Bit(out).nullaryExpr(f); in operator ()()
87 To32Bit(out).device(d) = To32Bit(out).constant(T(0)); in operator ()()
107 To32Bit(out).device(d) = To32Bit(out).constant(T(1)); in operator ()()
|
| D | constant_op_gpu.cu.cc | 74 To32Bit(out).device(d) = To32Bit(out).nullaryExpr(f); in operator ()()
87 To32Bit(out).device(d) = To32Bit(out).constant(T(0)); in operator ()()
100 To32Bit(out).device(d) = To32Bit(out).constant(T(1)); in operator ()()
|
| D | cwise_op_gpu_select.cu.cc | 32 To32Bit(out).device(d) = in operator ()()
33 To32Bit(cond_flat).select(To32Bit(then_flat), To32Bit(else_flat)); in operator ()()
51 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) + in operator()
47 To32Bit(bias).broadcast(bcast).reshape(one_d); 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 | scan_ops.h | 38 To32Bit(out).device(d) = in operator()
39 To32Bit(in).reverse(dims).scan(1, reducer, exclusive).reverse(dims); in operator()
|
| D | multinomial_op_gpu.cu.cc | 95 To32Bit(scores).device(d) = in operator ()()
96 To32Bit(logits).reshape(boc).broadcast(oso).template cast<float>() - in operator ()()
97 ((-((To32Bit(noises) + 2e-30f).log())).log()); in operator ()()
|
| 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 | 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 | fill_functor.cc | 68 To32Bit(out).device(d) = To32Bit(out).constant(T(0)); in operator ()()
155 To32Bit(out).device(d) = in.reshape(rank1).broadcast(broadcast_dims); in operator ()()
|
| D | pad_op.h | 38 To32Bit(output).device(d) = To32Bit(input).pad(paddings, pad_value); 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 | 39 To32Bit(output).device(d) = To32Bit(input).slice(slice_indices, slice_sizes); in operator ()()
48 To32Bit(output).device(d) = To32Bit(input).slice(slice_indices, slice_sizes); in operator ()()
|
| D | fused_batch_norm_op.cu.cc | 71 To32Bit(out).device(d) = in operator ()()
72 To32Bit(out).constant(Eigen::NumTraits<T>::quiet_NaN()); 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()
|
| D | broadcast_to_op.h | 38 To32Bit(out).device(device) = To32Bit(in).broadcast(bcast); in DoBCast32Bit()
|
| D | tile_functor.h | 45 To32Bit(y).device(d) = To32Bit(x).broadcast(b); in TileUsingEigen()
|
| D | topk_op_gpu.h | 446 auto input_indices_t = To32Bit(input_indices.flat<int32>());
525 To32Bit(indices).device(d) =
526 To32Bit(temp_indices.matrix<int32>()).slice(slice_indices, slice_sizes);
527 To32Bit(values).device(d) =
528 To32Bit(temp_values.matrix<T>()).slice(slice_indices, slice_sizes);
|
| D | conv_ops_3d.cc | 318 ctx->eigen_device<GPUDevice>(), To32Bit(input_param.tensor<T, 5>()), in launch()
320 To32Bit(transformed_input.tensor<T, 5>()), data_format); in launch()
392 To32Bit(filter.tensor<T, 5>()), in launch()
393 To32Bit(transformed_filter.tensor<T, 5>())); in launch()
|
| D | reduction_gpu_kernels.cu.h | 908 FillIdentityEigenImpl(d, To32Bit(out), reducer);
933 FillIdentityEigenImpl(d, To32Bit(out), reducer);
969 FillIdentityEigenImpl(d, To32Bit(out), reducer);
1012 FillIdentityEigenImpl(d, To32Bit(out), reducer);
1032 FillIdentityEigenImpl(d, To32Bit(out), reducer);
1052 FillIdentityEigenImpl(d, To32Bit(out), reducer);
1072 FillIdentityEigenImpl(d, To32Bit(out), reducer);
1092 FillIdentityEigenImpl(d, To32Bit(out), reducer);
1111 FillIdentityEigenImpl(d, To32Bit(out), reducer);
|