| /external/tensorflow/tensorflow/contrib/learn/python/learn/ops/ |
| D | losses_ops.py | 37 def mean_squared_error_regressor(tensor_in, labels, weights, biases, name=None): argument
49 def softmax_classifier(tensor_in, argument
|
| D | embeddings_ops.py | 73 def categorical_variable(tensor_in, n_classes, embedding_size, name): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | maxpooling_op.cc | 60 Tensor* input_backprop, const Tensor& tensor_in, const Tensor& out_backprop, in SpatialMaxPoolWithArgMaxHelper()
247 const Tensor& tensor_in = context->input(0); in Compute() local
325 const std::vector<int32>& stride, Padding padding, const Tensor* tensor_in, in MaxPoolingBackwardCustomKernel()
379 const Tensor& tensor_in = context->input(0); in Compute() local
485 const Tensor& tensor_in = context->input(0); in Compute() local
539 const Tensor& tensor_in, const Tensor& tensor_out, in SpatialMaxPoolGradGrad()
682 const Tensor& tensor_in = context->input(0); in Compute() local
781 const Tensor& tensor_in = context->input(0); in Compute() local
836 const Tensor& tensor_in = context->input(0); in Compute() local
922 const Tensor& tensor_in = context->input(0); in Compute() local
[all …]
|
| D | pooling_ops_common.h | 110 const Tensor& tensor_in = context->input(0); in Compute() local
147 const Tensor& tensor_in, const PoolParameters& params) { in DepthwiseMaxPool()
157 const Tensor& tensor_in, const PoolParameters& params, in SpatialMaxPool()
|
| D | cudnn_pooling_gpu.cc | 40 const Tensor& tensor_in, Tensor* output) { in Compute()
124 const Tensor* tensor_in, const Tensor* tensor_out, Tensor* input_backprop) { in Compute()
|
| D | quantized_pooling_ops.cc | 56 const Tensor& tensor_in = context->input(0); in Compute() local
|
| D | pooling_ops_common.cc | 158 TensorFormat data_format, const Tensor& tensor_in, in Compute()
288 Padding padding, TensorFormat data_format, const Tensor* tensor_in, in Compute()
|
| D | fractional_max_pool_op.cc | 78 const Tensor& tensor_in = context->input(0); in Compute() local
230 const Tensor& tensor_in = context->input(0); in Compute() local
|
| D | pooling_ops_3d.cc | 167 const Tensor& tensor_in = context->input(0); in Compute() local
340 const Tensor& tensor_in = context->input(0); in Compute() local
684 const Tensor& tensor_in = context->input(0); in Compute() local
|
| D | avgpooling_op.cc | 77 const Tensor& tensor_in = context->input(0); in Compute() local
142 const Tensor& tensor_in = context->input(0); in Compute() local
|
| D | mkl_maxpooling_op.cc | 79 const Tensor& tensor_in = MklGetInput(context, 0); in Compute() local
237 const Tensor& tensor_in = MklGetInput(context, 0); in Compute() local
391 const Tensor& tensor_in = MklGetInput(context, 0); in MklPrepareInputs() local
|
| D | fractional_avg_pool_op.cc | 75 const Tensor& tensor_in = context->input(0); in Compute() local
|
| D | mkl_avgpooling_op.cc | 70 const Tensor& tensor_in = MklGetInput(context, 0); in Compute() local
|