| /external/tensorflow/tensorflow/python/ops/ |
| D | random_ops.py | 74 shape_tensor = _ShapeTensor(shape)
79 shape_tensor, dtype, seed=seed1, seed2=seed2)
124 shape_tensor = _ShapeTensor(shape)
131 shape_tensor,
173 shape_tensor = _ShapeTensor(shape)
178 shape_tensor, dtype, seed=seed1, seed2=seed2)
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/testutil/ |
| D | batch_features_testutil.cc | 74 auto shape_tensor = Tensor(tensorflow::DT_INT64, {2}); in RandomlyInitializeBatchFeatures() local
75 tensorflow::test::FillValues<int64>(&shape_tensor, {batch_size, 1}); in RandomlyInitializeBatchFeatures()
76 sparse_float_feature_shapes_list.push_back(shape_tensor); in RandomlyInitializeBatchFeatures()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | transpose_conv.cc | 74 const TfLiteTensor* shape_tensor, in ResizeOutputTensor() argument
77 if (shape_tensor->type != kTfLiteInt32) { in ResizeOutputTensor()
79 shape_tensor->type); in ResizeOutputTensor()
83 TfLiteIntArray* shape = TfLiteIntArrayCreate(NumElements(shape_tensor)); in ResizeOutputTensor()
85 shape->data[i] = GetTensorData<int32_t>(shape_tensor)[i]; in ResizeOutputTensor()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | parameterized_truncated_normal_op.cc | 329 const Tensor& shape_tensor = ctx->input(0); in Compute() local
336 ctx, TensorShapeUtils::IsVector(shape_tensor.shape()), in Compute()
338 shape_tensor.shape().DebugString())); in Compute()
339 int32 num_batches = shape_tensor.flat<int32>()(0); in Compute()
342 const int32 num_dims = shape_tensor.dim_size(0); in Compute()
344 samples_per_batch *= shape_tensor.flat<int32>()(i); in Compute()
349 auto shape_vec = shape_tensor.flat<int32>(); in Compute()
|
| D | broadcast_to_op.cc | 44 const Tensor& shape_tensor = ctx->input(1); in Compute() local
48 ctx->op_kernel().MakeShape(shape_tensor, &output_shape)); in Compute()
|
| D | sparse_conditional_accumulator.h | 442 Tensor* shape_tensor; in ReturnShapeTensor() local
444 ctx, ctx->allocate_output(2, {accum_val_dims}, &shape_tensor)); in ReturnShapeTensor()
449 shape_tensor->flat<int64>()(0) = in ReturnShapeTensor()
452 shape_tensor->flat<int64>()(i) = accum_val_->dim_size(i); in ReturnShapeTensor()
|
| /external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| D | scatter_nd_fuzz.cc | 56 Tensor shape_tensor(tensorflow::DT_INT32, in FuzzImpl() local
66 auto flat_shape = shape_tensor.flat<int32>(); in FuzzImpl()
124 {"shape", shape_tensor}}); in FuzzImpl()
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator.py | 280 def shape_tensor(self, name="shape_tensor"): member in LinearOperator
296 return linear_operator_util.shape_tensor(self.shape.as_list())
331 return linear_operator_util.shape_tensor(
334 return self.shape_tensor()[:-2]
371 return array_ops.size(self.shape_tensor())
410 return self.shape_tensor()[-1]
449 return self.shape_tensor()[-2]
|
| D | linear_operator_zeros.py | 199 self._num_rows = linear_operator_util.shape_tensor(
206 self._num_columns = linear_operator_util.shape_tensor(
224 self._batch_shape_arg = linear_operator_util.shape_tensor(
441 return math_ops.reduce_min(self.shape_tensor()[-2:])
|
| D | linear_operator_identity.py | 87 return math_ops.reduce_min(self.shape_tensor()[-2:])
276 self._num_rows = linear_operator_util.shape_tensor(
284 self._batch_shape_arg = linear_operator_util.shape_tensor(
618 self._num_rows = linear_operator_util.shape_tensor(
|
| /external/tensorflow/tensorflow/contrib/libsvm/kernels/ |
| D | decode_libsvm_op.cc | 125 Tensor* shape_tensor; in Compute() local
128 &shape_tensor)); in Compute()
129 auto shape = shape_tensor->flat<int64>(); in Compute()
|
| /external/tensorflow/tensorflow/contrib/coder/kernels/ |
| D | range_coder_ops_test.cc | 320 Tensor shape_tensor{DT_INT32, {3}}; in TEST_F() local
323 shape_tensor.flat<int32>().setValues({4, 6, 8}); in TEST_F()
327 Node* shape = test::graph::Constant(&g, shape_tensor); in TEST_F()
474 Tensor shape_tensor{DT_INT32, {shape.dims()}}; in CreateRangeDecodeFullBroadcastGraph() local
476 shape_tensor.flat<int32>()(i) = shape.dim_size(i); in CreateRangeDecodeFullBroadcastGraph()
482 .Input(test::graph::Constant(g, shape_tensor)) in CreateRangeDecodeFullBroadcastGraph()
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_low_rank_update_test.py | 285 shape_tensor = operator.shape_tensor().eval(feed_dict=feed_dict)
286 self.assertAllEqual([2, 3, 3], shape_tensor)
|
| D | linear_operator_composition_test.py | 200 self.assertAllEqual((2, 3, 5), operator.shape_tensor().eval())
217 (1, 2, 3, 5), operator.shape_tensor().eval(feed_dict=feed_dict))
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.linalg.-linear-operator-identity.pbtxt | 124 name: "shape_tensor"
125 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'shape_tensor\'], "
|
| D | tensorflow.linalg.-linear-operator-kronecker.pbtxt | 127 name: "shape_tensor"
128 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'shape_tensor\'], "
|
| D | tensorflow.linalg.-linear-operator-zeros.pbtxt | 123 name: "shape_tensor"
124 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'shape_tensor\'], "
|
| D | tensorflow.linalg.-linear-operator-block-diag.pbtxt | 127 name: "shape_tensor"
128 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'shape_tensor\'], "
|
| D | tensorflow.linalg.-linear-operator-lower-triangular.pbtxt | 123 name: "shape_tensor"
124 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'shape_tensor\'], "
|
| D | tensorflow.linalg.-linear-operator-scaled-identity.pbtxt | 128 name: "shape_tensor"
129 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'shape_tensor\'], "
|
| D | tensorflow.linalg.-linear-operator-full-matrix.pbtxt | 123 name: "shape_tensor"
124 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'shape_tensor\'], "
|
| D | tensorflow.linalg.-linear-operator-composition.pbtxt | 127 name: "shape_tensor"
128 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'shape_tensor\'], "
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.linalg.-linear-operator-inversion.pbtxt | 127 name: "shape_tensor"
128 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'shape_tensor\'], "
|
| D | tensorflow.linalg.-linear-operator-scaled-identity.pbtxt | 128 name: "shape_tensor"
129 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'shape_tensor\'], "
|
| D | tensorflow.linalg.-linear-operator-identity.pbtxt | 124 name: "shape_tensor"
125 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'shape_tensor\'], "
|