Searched defs:shape_tensor (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | broadcast_to_op.cc | 44 const Tensor& shape_tensor = ctx->input(1); in Compute() local
|
| D | parameterized_truncated_normal_op.cc | 329 const Tensor& shape_tensor = ctx->input(0); in Compute() local
|
| D | sparse_conditional_accumulator.h | 442 Tensor* shape_tensor; in ReturnShapeTensor() local
|
| /external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| D | scatter_nd_fuzz.cc | 56 Tensor shape_tensor(tensorflow::DT_INT32, in FuzzImpl() local
|
| /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
|
| /external/tensorflow/tensorflow/contrib/libsvm/kernels/ |
| D | decode_libsvm_op.cc | 125 Tensor* shape_tensor; in Compute() local
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_util.py | 109 def shape_tensor(shape, name=None): function
|
| D | linear_operator.py | 280 def shape_tensor(self, name="shape_tensor"): member in LinearOperator
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | transpose_conv.cc | 74 const TfLiteTensor* shape_tensor, in ResizeOutputTensor()
|
| /external/tensorflow/tensorflow/contrib/coder/kernels/ |
| D | range_coder_ops_test.cc | 320 Tensor shape_tensor{DT_INT32, {3}}; in TEST_F() local
474 Tensor shape_tensor{DT_INT32, {shape.dims()}}; in CreateRangeDecodeFullBroadcastGraph() local
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| D | vector_diffeomixture.py | 869 def _set_event_shape(shape, shape_tensor): argument
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_old_batch_norms_test.cc | 390 Tensor shape_tensor(DT_INT32, TensorShape({})); in TestFoldFusedBatchNormsWithConcat() local
|
| D | quantize_nodes_test.cc | 371 Tensor shape_tensor(DT_INT32, TensorShape({})); in TestQuantizeConcat() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1639 const Tensor* shape_tensor = c->input_tensor(2); in SparseReduceShapeFn() local
|