OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:axes_shape
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
reverse_op.cc
78
const TensorShape
axes_shape
= ctx->InputShape(1);
variable
80
OP_REQUIRES(ctx, TensorShapeUtils::IsVector(
axes_shape
),
82
axes_shape
.DebugString()));
83
OP_REQUIRES(ctx,
axes_shape
.num_elements() <= x_shape.dims(),
84
errors::InvalidArgument("axes ",
axes_shape
.DebugString(),
89
if (
axes_shape
.num_elements() == 0) {
/external/tensorflow/tensorflow/python/ops/
math_ops.py
[
all
...]
Completed in 1506 milliseconds