OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:depth_dim
(Results
1 - 6
of
6
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
eigen_softmax_test.cc
39
Eigen::array<int, 1>
depth_dim
;
local
40
depth_dim
[0] = 0;
49
input.maximum(
depth_dim
).eval().reshape(dims2d).broadcast(bcast)) *
54
(reference.sum(
depth_dim
).eval().reshape(dims2d).broadcast(bcast));
eigen_softmax.h
49
Eigen::array<typename internal::traits<Input>::Index, 1>
depth_dim
;
local
50
depth_dim
[0] = 0;
60
Eigen::IndexList<Eigen::type2index<0> >
depth_dim
;
71
input.maximum(
depth_dim
).eval().reshape(dims2d).broadcast(bcast)) *
76
(output.sum(
depth_dim
).eval().reshape(dims2d).broadcast(bcast));
softmax_op_functor.h
60
Eigen::IndexList<Eigen::type2index<1> >
depth_dim
;
local
/external/eigen/unsupported/test/
cxx11_tensor_forced_eval.cpp
56
Eigen::array<int, 1>
depth_dim
;
local
57
depth_dim
[0] = 0;
65
Tensor<float, 2> output_tensor= (input_tensor - input_tensor.maximum(
depth_dim
).eval().reshape(dims2d).broadcast(bcast));
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
pooling_ops.cc
424
int
depth_dim
= GetTensorFeatureDimIndex(num_dims(), data_format_);
variable
425
int64 depth = out_backprop_shape.dim_size(
depth_dim
);
/external/tensorflow/tensorflow/core/framework/
common_shape_fns.cc
672
DimensionHandle
depth_dim
= c->Dim(
local
690
{output_rows, output_cols},
depth_dim
,
[
all
...]
Completed in 199 milliseconds