Searched defs:shape_tensor (Results 1 - 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/ops/linalg/
H A Dlinear_operator_util.py292 def shape_tensor(shape, name=None): function
H A Dlinear_operator.py288 def shape_tensor(self, name="shape_tensor"): member in class:LinearOperator
306 self._cached_shape_tensor = linear_operator_util.shape_tensor(
344 self._cached_batch_shape_tensor = linear_operator_util.shape_tensor(
347 self._cached_batch_shape_tensor = self.shape_tensor()[:-2]
387 self._cached_tensor_rank_tensor = array_ops.size(self.shape_tensor())
425 self._cached_domain_dimension_tensor = self.shape_tensor()[-1]
463 self._cached_range_dimension_tensor = self.shape_tensor()[-2]
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/testutil/
H A Dbatch_features_testutil.cc74 auto shape_tensor = Tensor(tensorflow::DT_INT64, {2}); local
75 tensorflow::test::FillValues<int64>(&shape_tensor, {batch_size, 1});
76 sparse_float_feature_shapes_list.push_back(shape_tensor);
/external/tensorflow/tensorflow/contrib/libsvm/kernels/
H A Ddecode_libsvm_op.cc125 Tensor* shape_tensor; variable
128 &shape_tensor));
129 auto shape = shape_tensor->flat<int64>();
/external/tensorflow/tensorflow/core/kernels/
H A Dparameterized_truncated_normal_op.cc254 const Tensor& shape_tensor = ctx->input(0); variable
261 ctx, TensorShapeUtils::IsVector(shape_tensor.shape()),
263 shape_tensor.shape().DebugString()));
264 int32 num_batches = shape_tensor.flat<int32>()(0);
267 const int32 num_dims = shape_tensor.dim_size(0);
269 samples_per_batch *= shape_tensor.flat<int32>()(i);
274 auto shape_vec = shape_tensor.flat<int32>();
H A Dsparse_conditional_accumulator.h442 Tensor* shape_tensor; local
444 ctx, ctx->allocate_output(2, {accum_val_dims}, &shape_tensor));
449 shape_tensor->flat<int64>()(0) =
452 shape_tensor->flat<int64>()(i) = accum_val_->dim_size(i);

Completed in 720 milliseconds