HomeSort by relevance Sort by last modified time
    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