OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:value_shape
(Results
1 - 4
of
4
) sorted by null
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
fill_op.cc
36
const TensorShape
value_shape
= ctx->InputShape(1);
variable
41
OP_REQUIRES(ctx, IsLegacyScalar(
value_shape
),
43
value_shape
.DebugString()));
60
if (
value_shape
.dims() > 0) {
61
CHECK_EQ(
value_shape
.dims(), 1);
tensor_array_ops.cc
277
std::vector<int64>
value_shape
(slice_shape.begin() + 1, slice_shape.end());
278
ctx->SetOutput(0, b->Reshape(read,
value_shape
));
365
const TensorShape
value_shape
= ctx->InputShape(2);
variable
369
TensorShape elem_shape =
value_shape
;
389
if (status.ok() && num_indices ==
value_shape
.dim_size(0)) {
402
auto slice_dims =
value_shape
.dim_sizes();
405
std::vector<int64> value_starts(
value_shape
.dims(), 0);
406
auto value_ends =
value_shape
.dim_sizes();
408
std::vector<int64> value_strides(
value_shape
.dims(), 1);
499
TensorShape
value_shape
= ctx->InputShape(1)
variable
[
all
...]
/external/tensorflow/tensorflow/core/ops/
resource_variable_ops.cc
97
ShapeHandle
value_shape
= c->input(1);
local
100
c->Merge(handle_shape_and_type.shape,
value_shape
, &unused));
/external/tensorflow/tensorflow/core/kernels/
lookup_table_op.cc
125
TensorShape
value_shape
() const override { return TensorShape(); }
154
GetNodeAttr(kernel->def(), "
value_shape
", &value_shape_));
254
TensorShape
value_shape
() const override { return value_shape_; }
310
GetNodeAttr(kernel->def(), "
value_shape
", &value_shape_));
472
TensorShape
value_shape
= MaybeVectorizeShape(value_shape_);
variable
479
expected_value_shape.AppendShape(
value_shape
);
494
TensorShape
value_shape
() const override { return value_shape_; }
673
output_shape.AppendShape(table->
value_shape
());
[
all
...]
Completed in 369 milliseconds